Nhận các chỉ số của giá trị True trong danh sách boolean

Tôi có một đoạn mã để tạo tổng đài. Tôi muốn trả lại danh sách tất cả các công tắc đang bật. Ở đây "on" sẽ bằng nhau Truevà "off" bằng nhau False. Vì vậy, bây giờ tôi chỉ muốn trả về một danh sách tất cả các Truegiá trị và vị trí của chúng. Đây là tất cả những gì tôi có nhưng nó chỉ trả về vị trí của lần xuất hiện đầu tiên True(đây chỉ là một phần mã của tôi):

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]

Điều này chỉ trả về "4"

Sử dụng enumerate, list.indextrả về chỉ mục của kết quả phù hợp đầu tiên được tìm thấy.

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

Đối với các danh sách lớn, tốt hơn nên sử dụng 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

Nếu bạn có sẵn 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 : sử dụng np.wherevì nó là lựa chọn nhanh nhất. Lựa chọn của bạn np.where, itertools.compressvà list comprehension.

Xem so sánh chi tiết bên dưới, nơi nó có thể được thấy np.wheretốt hơn cả hai itertools.compressvà cũng như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

• Phương pháp 1: Sử dụng 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)

• Phương pháp 2: Sử dụng itertools.compress

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

• Phương pháp 3 (phương pháp nhanh nhất): Sử dụng numpy.where

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

Bạn có thể sử dụng bộ lọc cho nó:

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

Các rangeđây liệt kê các yếu tố của danh sách của bạn và kể từ khi chúng tôi muốn chỉ những nơi self.statesđược True, chúng tôi đang áp dụng một bộ lọc dựa trên tình trạng này.

Đối với Python> 3.0:

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

Sử dụng cách hiểu từ điển,

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

Đầu vào:

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

Đầu ra:

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

Sử dụng phép nhân khôn ngoan nguyên tố và một tập hợp:

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

Đầu ra: {4, 5, 7}

Đơn giản chỉ cần làm điều này:

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