как найти все индекс элемента в списке Python

Question

Если у меня есть список

a=[1,0,0,1,0,1,1,1,0,1,0,0] 

Я хочу найти индекс 0 и 1 соответственно, скажем, в этом случае,

index_0 = [1,2,4,8,10,11] 
index_1 = [0,3,5,6,7,9] 

есть эффективный способ сделать это?

Answer 1

index_0 = [i for i, v in enumerate(a) if v == 0] 
index_1 = [i for i, v in enumerate(a) if v == 1] 

Или с NumPy:

import numpy as np 
a = np.array(a) 
index_0 = np.where(a == 0)[0] 
index_1 = np.where(a == 1)[0] 

Answer 2

Другой способ сделать это:

import os 

a = [1,0,0,1,0,1,1,1,0,1,0,0] 
index_0 = [] 
index_1 = [] 
aux = 0 

for i in a: 
    if i == 0: 
     index_0.append(aux) 
     aux += 1 
    else: 
     index_1.append(aux) 
     aux += 1 

print index_0 
print index_1 

Answer 3

с помощью itertools.compress:

>>> a=[1,0,0,1,0,1,1,1,0,1,0,0] 
>>> index_1 = [x for x in itertools.compress(range(len(a)),a)] 
>>> index_1 
[0, 3, 5, 6, 7, 9] 
>>> index_0 = [x for x in itertools.compress(range(len(a)),map(lambda x:not x,a))] 
>>> index_0 
[1, 2, 4, 8, 10, 11] 

можно достичь, используя один цикл: для повышения эффективности и эффективности

>>> a=[1,0,0,1,0,1,1,1,0,1,0,0] 
>>> index_0 = [] 
>>> index_1 = [] 
>>> for i,x in enumerate(a): 
...  if x: index_1.append(i) 
...  else: index_0.append(i) 
... 
>>> index_0 
[1, 2, 4, 8, 10, 11] 
>>> index_1 
[0, 3, 5, 6, 7, 9]