cumsum() :连加函数
- numpy.cumsum(a, axis=None, dtype=None, out=None)
axis=0,按照行累加。
axis=1,按照列累加。
axis不给定具体值,就把numpy数组当成一个一维数组。
例子
a= np.array([[1,2,3],[4,5,6]])
np.cumsum(a)
#结果: 默认是按顺序全部连加,相当于一个一维数组
array([ 1, 3, 6, 10, 15, 21], dtype=int32)
np.cumsum(a,axis=0)
结果:axis=0 是按行连加,第一次是第一行1,2,3
第二次是第二行加上第一行 为5,7,9
array([[1, 2, 3],
[5, 7, 9]], dtype=int32)
np.cumsum(a,axis=1)
结果:第一次为第一列1,4。第二次为第二列加上第一列为3,9,
第三次是第三列加上前两列为6,9
array([[ 1, 3, 6],
[ 4, 9, 15]], dtype=int32)
cumprod() :连乘跟连加一样
例子
np.cumprod(a)
结果:按顺序连乘,相当于一维数组
array([ 1, 2, 6, 24, 120, 720], dtype=int32)
np.cumprod(a,axis=0)
结果:第一次为第一行,第二次为第二行乘以第一行
array([[ 1, 2, 3],
[ 4, 10, 18]], dtype=int32)
np.cumprod(a,axis=1)
结果:#第一次为第一列,第二次为第二列乘以第一列,第三次为第三列乘以前两列
array([[ 1, 2, 6],
[ 4, 20, 120]], dtype=int32)