在布尔列表中获取True值的索引

Question

我有一段代码应该在其中创建总机。我想返回所有打开的开关的列表。这里的“开”等于True和“关”等于False。因此，现在我只想返回所有True值及其位置的列表。这就是我所拥有的，但它只返回第一次出现的位置True（这只是我的代码的一部分）：

self.states = [False, False, False, False, True, True, False, True, False, False, False, False, False, False, False, False]

def which_switch(self):
    x = [self.states.index(i) for i in self.states if i == True]

这只会返回“ 4”

Answer 1

使用enumerate，list.index返回找到的第一个匹配项的索引。

>>> t = [False, False, False, False, True, True, False, True, False, False, False, False, False, False, False, False]
>>> [i for i, x in enumerate(t) if x]
[4, 5, 7]

对于庞大的列表，最好使用itertools.compress：

>>> from itertools import compress
>>> list(compress(xrange(len(t)), t))
[4, 5, 7]
>>> t = t*1000
>>> %timeit [i for i, x in enumerate(t) if x]
100 loops, best of 3: 2.55 ms per loop
>>> %timeit list(compress(xrange(len(t)), t))
1000 loops, best of 3: 696 µs per loop