送你一朵野菊花
numpy.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]])
>>> a
array([[1, 2, 3],
[4, 5, 6]])
>>> np.cumsum(a)
array([ 1, 3, 6, 10, 15, 21])
>>>
>>> np.cumsum(a,axis=0) #按照行累加,行求和
array([[1, 2, 3],
[5, 7, 9]])
[1, 2, 3]------> |1 |2 |3 |
[4, 5, 6]------> |5=1+4 |7=2+5 |9=3+6|
>>> np.cumsum(a,axis=1) #按照列累加,列求和
array([[ 1, 3, 6],
[ 4, 9, 15]])
[1, 2, 3]------> |1 |2+1 |3+2+1 |
[4, 5, 6]------> |4 |4+5 |4+5+6 |
>>> np.cumsum(a, dtype=float) # 指定輸出類型。
注意啦!沒有指定軸參數(axis)!輸出就變成1維數組了。
array([ 1., 3., 6., 10., 15., 21.])第一步:每個值都變成float了
array([1,1+2=3,1+2+3=6,1+2+3+4=10,1+2+3+4+5=15,1+2+3+4+5+6=21])第二部:累加