我有一个总是只包含 1 和 0 的列表。 我需要获取列表的非零索引列表:
a = [0, 1, 0, 1, 0, 0, 0, 0]
b = []
for i in range(len(a)):
if a[i] == 1: b.append(i)
print b
实现这一目标的“pythonic”方式是什么?
最佳答案
[i for i, e in enumerate(a) if e != 0]
关于list - 如何获取列表中非零元素的索引列表?,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/4111412/