Есть ли эффективный механизм Numpy для извлечения целочисленных индексов местоположений в массиве на основе условия, это правда, в отличие от массива логических масок?
Например:
x=np.array([range(100,1,-1)])
#generate a mask to find all values that are a power of 2
mask=x&(x-1)==0
#This will tell me those values
print x[mask]
В этом случае я хотел бы знать индексы i
of mask
, где mask[i]==True
. Можно ли генерировать их без циклов?