cumsum()函数

1、 对于一维输入a（可以是list，可以是array，假设a=[1, 2, 3, 4, 5, 6,] ，就是当前列之前的和加到当前列上，如下：

import numpy as np
a = [1,2,3,4,5,6]
print( np.cumsum(a))

结果为：

[1,   3,   6,  10,  15,  21,  28,  36,  45,  55,  75]

---

对于二维输入a，axis=0(第1行不动，将第1行累加到其他行)；axis=1(进入最内层，转化成列处理。第1列不动，将第1列累加到其他列)，如下：

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

L2 结果为：

[[ 1  2  3]
 [ 5  7  9]
 [12 15 18]]

L3 结果为：

[[ 1  3  6]
 [ 4  9 15]
 [ 7 15 24]]