numpyの行列マイナス繰り返し行/列のインデックス

Question

私が欲しいもの

>>> a 
array([[1, 2], 
    [3, 4], 
    [5, 6]]) 
>>>b 
array([[1, 2], 
    [4, 3]]) 

とベクトルは

b[0,:]-=a[0,:], b[1,:]-=a[1,:], b[0,:]-=a[2,:] 

を行っている。しかし、次は動作しません：

>>> b[[0,1,0],:]-=a 
>>> b 
array([[-4, -4], # want [-5, -6] 
    [ 1, -1]]) 

これがすることなく達成することができる方法forループを使用していますか？

Answer 1

使用np.subtract.at -

np.subtract.at(b,[0,1,0],a) 

サンプル実行 -

In [15]: a 
Out[15]: 
array([[1, 2], 
     [3, 4], 
     [5, 6]]) 

In [16]: b 
Out[16]: 
array([[1, 2], 
     [4, 3]]) 

In [17]: np.subtract.at(b,[0,1,0],a) 

In [18]: b 
Out[18]: 
array([[-5, -6], 
     [ 1, -1]])