numpy.cumsum()

numpy.cumsum() 是 NumPy 中的一个函数，它用于返回给定数组的逐元素累计和。

在 numpy.cumsum() 中，cumulative 表示“累计的”，sum 表示“和”，所以 numpy.cumsum() 的功能就是返回给定数组的累计和。

语法

numpy.cumsum(a, axis=None, dtype=None, out=None)

其中：

• a：给定的数组。
• axis：指定累加和的方向，如果不指定方向，则返回累加和的结果。默认为 None。
• dtype：输出数组的类型。
• out：输出数组的位置。

示例

import numpy as np

a = np.array([1, 2, 3, 4, 5, 6])
b = np.cumsum(a)

print(b)

输出：

[ 1  3  6 10 15 21]

解释

在这个例子中，我们给 np.cumsum() 传递了一个数组 a，并且返回一个数组 b，其中 b 的每个元素都是原始数组 a 的累计和。

指定方向

import numpy as np

a = np.array([[ 0,  1,  2],
              [ 3,  4,  5],
              [ 6,  7,  8],
              [ 9, 10, 11]])

b = np.cumsum(a, axis=0)

print(b)

输出：

[[ 0  1  2]
 [ 3  5  7]
 [ 9 12 15]
 [18 22 26]]

解释

在这个例子中，我们传递了一个多维数组 a 给 numpy.cumsum()，并通过 axis 参数指定了沿着列的方向进行累计和。结果数组 b 的每个元素都是原始数组 a 沿着列的累计和。