如何将布尔数组转换为numpy中的索引数组-问答-阿里云开发者社区-阿里云

开发者社区> 问答> 正文

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

保持可爱mmm 2020-02-09 13:12:00 99

有没有一种有效的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。是否可以生成这些而不循环? 问题来源于stack overflow

索引 Python
分享到
取消 提交回答
全部回答(1)
  • 保持可爱mmm
    2020-02-09 13:12:12

    另外的选择:

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

    0 0
开发与运维
使用钉钉扫一扫加入圈子
+ 订阅

集结各类场景实战经验,助你开发运维畅行无忧

推荐文章
相似问题
推荐课程