是否有一个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])
答案 0 :(得分:4)
v[b1 & b2]
会做到这一点。