有没有一种高效的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]
在这种情况下,我想知道 mask
中 i
的索引,使得 mask[i]==True
。有没有不用循环就能生成这些索引的方法?
numpy.flatnonzero(mask)
。 - christianbrodbeck