语法:numpy.cumsum(a, axis=None, dtype=None, out=None)
按照所给定的轴参数(axis)返回元素的梯形累计和,axis=0,按照列累加。axis=1,按照行累加。axis不给定具体值,就把numpy数组当成一个一维数组
举个简单例子(二维)
import numpy as np
a=np.arange(0,12).reshape(3,4)
print(a)
# [[ 0 1 2 3]
# [ 4 5 6 7]
# [ 8 9 10 11]]
b=np.cumsum(a)
print(b)
#[ 0 1 3 6 10 15 21 28 36 45 55 66]
不输入axis参数时当作一维数组累加:
0=0
1=0+1
3=0+1+2
6=0+1+2+3
以此类推
axis=0
b=np.cumsum(a,axis=0)
print(b)
#[[ 0 1 2 3]
#[ 4 6 8 10]
#[12 15 18 21]]
第一行:0=0,1=1,2=2,3=3
第二行:4=0+4,6=1+5,8=2+6,10=3+7
以此类推
axis=1
b=np.cumsum(a,axis=1)
print(b)
#[[ 0 1 3 6]
#[ 4 9 15 22]
#[ 8 17 27 38]]