Bir boole listesindeki True değerlerinin indislerini alma

Santral oluşturmam gereken kodumun bir parçası var. Açık olan tüm anahtarların bir listesini döndürmek istiyorum. Burada "açık" eşit olacak Trueve "kapalı" eşit olacaktır False. Şimdi sadece tüm Truedeğerlerin ve konumlarının bir listesini döndürmek istiyorum . Sahip olduğum tek şey bu, ancak yalnızca ilk oluşumunun konumunu döndürüyor True(bu, kodumun yalnızca bir kısmı):

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]

Bu yalnızca "4" değerini döndürür

Kullan enumerate, list.indexbulunan ilk eşleşmenin dizinini döndürür.

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

Büyük listeler için itertools.compressşunları kullanmak daha iyi olacaktır :

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

Eğer uyuşturucunuz varsa:

>>> 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.whereen hızlı seçenek olarak kullanın. Seçenekleriniz şunlardır np.where, itertools.compressve list comprehension.

np.whereHem itertools.compressve hem de daha iyi performans gösterdiğininlist comprehension görülebileceği aşağıdaki ayrıntılı karşılaştırmaya bakın .

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

• Yöntem 1: Kullanma 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)

• Yöntem 2: Kullanma itertools.compress

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

• Yöntem 3 (en hızlı yöntem): Kullanma numpy.where

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

Bunun için filtre kullanabilirsiniz:

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

rangeBuraya listenin elemanlarını sıralar ve biz burada sadece bu istiyoruz çünkü self.statesedilir True, bu rahatsızlıklarda temel alan bir filtre uygulamaktadır.

Python> 3.0 için:

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

Sözlük anlama yolu kullanın,

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

Giriş:

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

Çıktı:

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

Eleman bazlı çarpma ve bir küme kullanma:

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

Çıktı: {4, 5, 7}

Basitçe şunu yapın:

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