只为一列累积numpy数组

时间:2011-03-03 05:20:44

标签: python numpy

我有一个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中执行此操作?

1 个答案:

答案 0 :(得分:2)

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

a现在是:

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