Jeu d'index Python numpy à partir d'un tableau booléen

Question

Comment transformer un tableau booléen en un itérable d'index?

Par ex,

import numpy as np 
import itertools as it 
x = np.array([1,0,1,1,0,0]) 
y = x > 0 
retval = [i for i, y_i in enumerate(y) if y_i] 

Y at-il une façon plus agréable?

Answer 1

Essayez np.where ou np.nonzero.

x = np.array([1, 0, 1, 1, 0, 0]) 
np.where(x)[0] # returns a tuple hence the [0], see help(np.where) 
# array([0, 2, 3]) 
x.nonzero()[0] # in this case, the same as above. 

Voir help(np.where) et help(np.nonzero).

Peut-être noter que dans la page np.where il mentionne que pour 1D x il est fondamentalement équivalent à votre longform dans la question.