Pobieranie indeksów wartości True na liście boolowskiej

Mam fragment mojego kodu, w którym mam utworzyć centralę. Chcę zwrócić listę wszystkich włączonych przełączników. Tutaj „włączony” będzie równy, Truea „wyłączony” równy False. Teraz chcę po prostu zwrócić listę wszystkich Truewartości i ich pozycji. To wszystko, co mam, ale zwraca tylko pozycję pierwszego wystąpienia True(to tylko część mojego kodu):

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]

Zwraca tylko „4”

Użyj enumerate, list.indexzwraca indeks pierwszego znalezionego dopasowania.

>>> 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]

W przypadku dużych list lepiej byłoby użyć 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

Jeśli masz dostępne 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 : użyj, np.whereponieważ jest to najszybsza opcja. Opcje są np.where, itertools.compressi list comprehension.

Zobacz szczegółowe porównanie poniżej, na którym widać, że np.whereprzewyższa zarówno itertools.compressi również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

• Metoda 1: Używanie 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)

• Metoda 2: Używanie itertools.compress

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

• Metoda 3 (najszybsza metoda): przy użyciu numpy.where

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

Możesz użyć do tego filtra:

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

rangeTutaj wylicza elementy listy i ponieważ chcemy tylko tych, gdzie self.statesjest True, jesteśmy stosowania filtru w oparciu o ten stan.

Dla Pythona> 3.0:

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

Korzystaj ze słownika,

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

Wejście:

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

Wynik:

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

Korzystanie z mnożenia elementów i zbioru:

>>> 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})

Wynik: {4, 5, 7}

Po prostu zrób to:

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