Отримання індексів True значень у логічному списку

У мене є частина мого коду, де я повинен створити комутатор. Я хочу повернути список усіх увімкнених комутаторів. Тут "on" буде рівним Trueі "off" рівним 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"

Use 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; ДР : використовуйте, 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
    ]