ブールリストでTrue値のインデックスを取得する

配電盤を作成することになっているコードの一部があります。オンになっているすべてのスイッチのリストを返したいのですが。ここで、「オン」は等しく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」のみを返します

を使用して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

numpyを利用できる場合：

>>> import numpy as np
>>> states = [False, False, False, False, True, True, False, True, False, False, False, False, False, False, False, False]
>>> np.where(states)[0]
array([4, 5, 7])

TL; DR：np.where最速のオプションなので使用してください。次のオプションがありnp.where、itertools.compressとlist comprehension。

以下の詳細な比較を参照してください。ここではnp.where、itertools.compressとの両方を上回っていlist comprehensionます。

>>> from itertools import compress
>>> import numpy as np
>>> t = [False, False, False, False, True, True, False, True, False, False, False, False, False, False, False, False]`
>>> t = 1000*t

• 方法1：使用 list comprehension

>>> %timeit [i for i, x in enumerate(t) if x]
457 µs ± 1.5 µs per loop (mean ± std. dev. of 7 runs, 1000 loops each)

• 方法2：使用 itertools.compress

>>> %timeit list(compress(range(len(t)), t))
210 µs ± 704 ns per loop (mean ± std. dev. of 7 runs, 1000 loops each)

• 方法3（最速の方法）：使用 numpy.where

>>> %timeit np.where(t)
179 µs ± 593 ns per loop (mean ± std. dev. of 7 runs, 10000 loops each)

あなたはそれのためにフィルターを使うことができます：

filter(lambda x: self.states[x], range(len(self.states)))

rangeここであなたのリストの要素を列挙し、私たちが望むだけなので、これらの場所self.statesでありますTrue、私たちはこの条件に基づいてフィルタを適用しているが。

Python> 3.0の場合：

list(filter(lambda x: self.states[x], range(len(self.states))))

辞書理解方法を使用して、

x = {k:v for k,v in enumerate(states) if v == True}

入力：

states = [False, False, False, False, True, True, False, True, False, False, False, False, False, False, False, False]

出力：

{4: True, 5: True, 7: True}

要素ごとの乗算とセットの使用：

>>> states = [False, False, False, False, True, True, False, True, False, False, False, False, False, False, False, False]
>>> set(multiply(states,range(1,len(states)+1))-1).difference({-1})

出力： {4, 5, 7}

単にこれを行う：

def which_index(self):
    return [
        i for i in range(len(self.states))
        if self.states[i] == True
    ]