Pythonプログラミングの練習Numpyの使い方

Numpyパッケージでは、ブール配列で値を取得できます.

>>> a = np.arange(12).reshape(3,4)
>>> b1 = np.array([False,True,True])             #   a    
>>> b2 = np.array([True,False,True,False])       #   a    
>>>
>>> a[b1,:]                                   #            
array([[ 4,  5,  6,  7],
       [ 8,  9, 10, 11]])
>>>
>>> a[b1]                                     # same thing
array([[ 4,  5,  6,  7],
       [ 8,  9, 10, 11]])
>>>
>>> a[:,b2]                                   #           
array([[ 0,  2],
       [ 4,  6],
       [ 8, 10]])
>>>
>>> a[b1,b2]                                  #       
array([ 4, 10])

テキストリンク