python累计值计算

numpy.cumsum() 函数的功能是返回给定axis上的累计和

一维：

import numpy as np
a=np.arange(6)
b=[0,1,2,3,4,5]
c=range(6)
print(type(a),type(b),type(c))
print(np.cumsum(a))
print(np.cumsum(b))
print(np.cumsum(c))

输出结果：

<class 'numpy.ndarray'> <class 'list'> <class 'range'>
[ 0  1  3  6 10 15]
[ 0  1  3  6 10 15]
[ 0  1  3  6 10 15]

二维：axis=0行，axis=1列

import numpy as np
a=[[1,2,3],[4,5,6],[7,8,9]]
print(type(a))
print(a)
print(np.cumsum(a,axis=0))
print(np.cumsum(a,axis=1))

输出结果：

<class 'list'>
[[1, 2, 3], [4, 5, 6], [7, 8, 9]]
[[ 1  2  3]
 [ 5  7  9]
 [12 15 18]]
[[ 1  3  6]
 [ 4  9 15]
 [ 7 15 24]]