накапливать массив всего за один столбец

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