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]])