2012-02-17 25 views
2

2つのブール値の配列の値のand関数に応じて配列の項目を返すnumpy関数(またはアルゴリズム)がありますか?特定のインデックスのみを持つnumpy配列の取得

など。

>>> 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文字未満の2番目の答えです:-) –

関連する問題