numpy.cumsum(a, axis=None, dtype=None, out=None)介绍。

主要是axis，按照所给定的轴参数（axis）返回元素的梯形累计和，axis=0，按照列累加。axis=1，按照行累加。axis不给定具体值，就把numpy数组当成一个一维数组。
image.png

1
2
3
import numpy as np
a = np.arange(1,7).reshape(2,3)
a
1
2
array([[1, 2, 3],
       [4, 5, 6]])
1
2
3
# 默认没有指定轴，所以输出会变成一维数组
# array([1，1+2=3，1+2+3=6，1+2+3+4=10，1+2+3+4+5=15，1+2+3+4+5+6=21]）
np.cumsum(a)
1
array([ 1,  3,  6, 10, 15, 21])
1
2
3
4
# [1,2,3]
# [1+4=5,2+5=7,3+6=9]
# 按照轴0进行累计求
np.cumsum(a,axis=0)
1
2
array([[1, 2, 3],
       [5, 7, 9]])
1
2
3
4
# 按照轴1进行累计求和
# [1,1+2=3,1+2+3=6]
# [4,4+5=9,4+5+6=15]
np.cumsum(a,axis=1)
1
2
array([[ 1,  3,  6],
       [ 4,  9, 15]])