如何在numpy中将布尔数组转换为索引数组

Question

是否有一个有效的Numpy机制来检索基于条件为true的数组中位置的整数索引,而不是布尔掩码数组？

例如:

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的mask地方mask[i]==True.是否有可能在没有循环的情况下生成这些？

Answer 1

另外一个选项:

In [13]: numpy.where(mask)
Out[13]: (array([36, 68, 84, 92, 96, 98]),)

这是一回事numpy.where(mask==True).

Answer 2

您应该可以使用它numpy.nonzero()来查找此信息.