gpt4 book ai didi

Python - 另一个列表中符合条件的元素的索引列表

转载 作者:行者123 更新时间:2023-11-28 21:03:43 27 4
gpt4 key购买 nike

这看起来很简单,但我有点卡在这上面了。
我有一个逻辑值列表,例如:

a=[True, False, True, True, False, False, True, True, True]

我想构建另一个列表,其中包含(在子列表中)为“True”的“a”的索引,连续索引放在同一个子列表中。所以,对于上面的例子,答案是:

[[0], [2,3], [6,7,8]] 

最佳答案

使用 itertools.groupbyenumerate , 一行是可能的:

from itertools import groupby
[[i for i, _ in g] for k, g in groupby(enumerate(a), key=lambda x: x[1]) if k]

关于Python - 另一个列表中符合条件的元素的索引列表,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/46247237/

27 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com