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

这里，. 表示数字是浮点数。