我有一个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中执行此操作?
答案 0 :(得分:2)
a = np.array([[1,2],[2,4]])
np.add.accumulate(a[:,1], out=a[:,1])
a
现在是:
array([[1, 2],
[2, 6]])