積累numpy數組只爲一列

Question

我有一個NumPy數組，我想積累一列的值，說2 ^nd列。

a = np.array([[1,2],[2,4]]) 
# some kind of accumulate function that accumulates just one column: 
np.add.accumulate(a, 2) 

a現在應該[[1,2],[2,6]]

有沒有辦法在NumPy的做到這一點？

Answer 1

a = np.array([[1,2],[2,4]]) 
np.add.accumulate(a[:,1], out=a[:,1]) 

a現在是：

array([[1, 2], 
     [2, 6]])