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]]