正对累计求和的应用场景来说,一个好的方法能节省大量的时间。
1、一维数组的累计求和:
a = np.random.randint(1,10,5)
out:array([6, 4, 5, 7, 2])
a.cumsum()
out:array([ 6, 10, 15, 22, 24])
2、二维数组的累计求和:
b = np.random.randint(1,10,(5,2))
out:array([[7, 3],
[1, 5],
[5, 2],
[5, 2],
[7, 3]])
print('b.cumsum(axis=0):',b.cumsum())
out:b.cumsum(axis=0): [ 7 10 11 16 21 23 28 30 37 40]