Abrufen von Indizes für True-Werte in einer Booleschen Liste

Ich habe einen Teil meines Codes, in dem ich eine Telefonzentrale erstellen soll. Ich möchte eine Liste aller eingeschalteten Schalter zurückgeben. Hier ist "Ein" gleich Trueund "Aus" gleich False. Jetzt möchte ich nur eine Liste aller TrueWerte und ihrer Position zurückgeben. Dies ist alles, was ich habe, aber es gibt nur die Position des ersten Auftretens von zurück True(dies ist nur ein Teil meines Codes):

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]

Dies gibt nur "4" zurück.

Verwenden enumerate, list.indexgibt den Index der ersten gefundenen Übereinstimmung zurück.

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

Für große Listen ist es besser, Folgendes zu verwenden 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

Wenn Sie numpy zur Verfügung haben:

>>> 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 : Verwenden Sie, np.whereda dies die schnellste Option ist. Ihre Optionen sind np.where, itertools.compressund list comprehension.

Siehe den Vergleich unten, wo es gesehen werden kann , np.whereübertrifft sowohl itertools.compressals auchlist 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

• Methode 1: Verwenden 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)

• Methode 2: Verwenden itertools.compress

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

• Methode 3 (die schnellste Methode): Verwenden numpy.where

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

Sie können Filter dafür verwenden:

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

Das rangehier listet Elemente Ihrer Liste auf und da wir nur diejenigen wollen, wo es self.statesist True, wenden wir einen Filter an, der auf dieser Bedingung basiert.

Für Python> 3.0:

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

Verwenden Sie Wörterbuch Verständnis Weg,

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

Eingang:

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

Ausgabe:

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

Verwenden der elementweisen Multiplikation und einer Menge:

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

Ausgabe: {4, 5, 7}

Mach das einfach:

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