numpy中的cumsum()函数的使用

关于numpy.cumsum()在官网中官方给了一个很好的例子。

cumsum的作用主要就是计算轴向的累加和。

里面最重要的参数就是axis

当里面的参数axis没有值时，默认是一个一维数组进行加和，即
a=np.array([[1,2,3],[4,5,6]])
np.cumsum()
输出1,3,6,10,15,21
即1,1+2,1+2+3,1+2+3+4,1+2+3+4+5,1+2+3+4+5+6

np.cumsum(axis=0)，按行累加
输出
array([[1, 2, 3],
[5, 7, 9]], dtype=int32)
5=1+4,7=2+5,9=3+6

np.cumsum(axis=1)，按列累加
输出
array([[ 1, 3, 6],
[ 4, 9, 15]], dtype=int32)
3=1+2，6=1+2+3
9=4+5，15=4+5+6

感谢阅读，
我是啊帅和和，一位大数据专业大三学生。