2012-02-17 18 views
2

是否存在一個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]) 

回答

4
v[b1 & b2] 

會做的伎倆。

+0

謝謝......這是我的問題的完美解決方案少於10個字符的第二個答案:-) – 2012-02-17 17:26:52