只使用某些索引获取numpy数组

时间:2012-02-17 17:19:58

标签: python numpy

是否有一个numpy函数(或算法)只返回数组的项目,具体取决于其他两个布尔数组的值和函数。

E.g。

>>> b1 = numpy.array([False, False, True, True , True])
>>> b2 = numpy.array([True , False, True, False, True])
>>> v  = numpy.array([2    , 4    , 6   , 8,     10  ])

然后该函数应返回:

numpy.array([6, 10])

因为6和10是相应的b1和b2值均为True的值。

根据下面接受的答案编辑:

>>> v[b1 & b2]
array([ 6, 10])

1 个答案:

答案 0 :(得分:4)

v[b1 & b2]

会做到这一点。