python numpy中cumsum的用法詳解

Cumsum ：計(jì)算軸向元素累加和，返回由中間結(jié)果組成的數(shù)組

重點(diǎn)就是返回值是“由中間結(jié)果組成的數(shù)組”

以下代碼在python3.6版本運(yùn)行成功！

下面看代碼，定義一個(gè)2*2*3的數(shù)組，所以其shape是2，2，3，索引分別0，1，2

代碼：

import numpy as np
arr = np.array([[[1,2,3],[8,9,12]],[[1,2,4],[2,4,5]]]) #2*2*3
print(arr.cumsum(0))
print(arr.cumsum(1))
print(arr.cumsum(2))

輸出結(jié)果:

#cumsum(0)
[[[ 1  2  3]
  [ 8  9 12]]
 
 [[ 2  4  7]
  [10 13 17]]]
#cumsum(1)
[[[ 1  2  3]
  [ 9 11 15]]
 
 [[ 1  2  4]
  [ 3  6  9]]]
#cumsum(2)
[[[ 1  3  6]
  [ 8 17 29]]
 
 [[ 1  3  7]
  [ 2  6 11]]]

注釋：

• arr是一個(gè)2*2*3三維矩陣，索引值為0，1，2
• cumsum(0)：實(shí)現(xiàn)0軸上的累加：以最外面的數(shù)組元素為單位，以[[1,2,3],[8,9,12]]為開(kāi)始實(shí)現(xiàn)后面元素的對(duì)應(yīng)累加
• cumsum(1)：實(shí)現(xiàn)1軸上的累加：以中間數(shù)組元素為單位，以[1,2,3]為開(kāi)始，實(shí)現(xiàn)后面元素的對(duì)應(yīng)累加
• cumsum(2)：實(shí)現(xiàn)2軸上的累加：以最里面的元素為累加單位，即1為開(kāi)始，實(shí)現(xiàn)后面的元素累加

四維數(shù)組實(shí)現(xiàn)

下面看一個(gè)四維數(shù)組2*2*2*4，索引值為0，1，2，3
代碼：

import numpy as np
arr = np.arange(32).reshape((2,2,2,4))
print(arr)
print(arr.cumsum(0))
print(arr.cumsum(1))
print(arr.cumsum(2))
print(arr.cumsum(3))

arr:
[[[[ 0 1 2 3]
  [ 4 5 6 7]]
 
 [[ 8 9 10 11]
  [12 13 14 15]]]
 
 
 [[[16 17 18 19]
  [20 21 22 23]]
 
 [[24 25 26 27]
  [28 29 30 31]]]]

arr是一個(gè)2*2*2*4四維矩陣，索引值為0，1，2，3

cumsum(0)：實(shí)現(xiàn)0軸上的累加即：以最外面數(shù)組元素為單位即

[[[ 0 1 2 3]
  [ 4 5 6 7]]
 
 [[ 8 9 10 11]
  [12 13 14 15]]]

與

[[[16 17 18 19]
  [20 21 22 23]]
 
 [[24 25 26 27]
  [28 29 30 31]]]]

對(duì)應(yīng)位置元素相加起來(lái)

結(jié)果：

[[[[ 0 1 2 3]
  [ 4 5 6 7]]
 
 [[ 8 9 10 11]
  [12 13 14 15]]]
 
 
 [[[16 18 20 22]
  [24 26 28 30]]
 
 [[32 34 36 38]
  [40 42 44 46]]]]

cumsum(1)：實(shí)現(xiàn)1軸上的累加即：以次外面元素為單位，累加：

[[ 0 1 2 3]
  [ 4 5 6 7]]

加

 [[ 8 9 10 11]
  [12 13 14 15]]

加

[[16 17 18 19]
  [20 21 22 23]]

加

[[24 25 26 27]
  [28 29 30 31]]

累計(jì)過(guò)程產(chǎn)生的中間結(jié)果要記錄到數(shù)組中

所以，結(jié)果：

[[[[ 0 1 2 3]
  [ 4 5 6 7]]
 
 [[ 8 10 12 14]     
  [16 18 20 22]]]
 
 
 [[[16 17 18 19]
  [20 21 22 23]]
 
 [[40 42 44 46]
  [48 50 52 54]]]]

cumsum(2)就對(duì)應(yīng)從[ 0  1  2  3]數(shù)組元素開(kāi)始實(shí)現(xiàn)累加，然后記錄中間結(jié)果

cumsum(3)對(duì)應(yīng)的是從最里面最小的數(shù)組元素，即從0開(kāi)始實(shí)現(xiàn)累加，記錄中間結(jié)果

以上就是本文的全部?jī)?nèi)容，希望對(duì)大家的學(xué)習(xí)有所幫助，也希望大家多多支持億速云。