Python NumPy cumsum方法用法及代碼示例

例子

一維數組

計算一維數組的累積和：

x = np.array([1,2,3])
np.cumsum(x)



array([1, 3, 6])

在這裏，我們正在執行以下計算：

[0] 1 = 1
[1] 1 + 2 = 3
[2] 1 + 2 + 3 = 6

二維數組

考慮以下二維數組：

x = np.array([[1,2], [3,4]])
x



array([[1, 2],
       [3, 4]])

所有值

要計算所有值的累積和：

x = np.array([[1,2], [3,4]])
np.cumsum(x)



array([ 1,  3,  6, 10])

按列

要按列計算累積和，請設置 axis=0 ：

x = np.array([[1,2], [3,4]])
np.cumsum(x, axis=0)



array([[1, 2],
       [4, 6]])

逐行

要按行計算累積和，請設置 axis=1 ：

x = np.array([[1,2], [3,4]])
np.cumsum(x, axis=1)



array([[1, 3],
       [3, 7]])

指定數據類型

要獲取數據類型 float 的數組：

x = np.array([1,2,3])
np.cumsum(x, dtype=float)



array([1., 3., 6.])

這裏，. 表示數字是浮點數。