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

小编典典

python

有没有一种有效的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。是否可以生成这些而不循环？

阅读 318

2021-01-20

共1个答案

小编典典

另外的选择：

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

这与numpy.where(mask==True)。

2021-01-20