cumsum函数如何使用

cumsum函数是numpy库中的一个函数，可以用于计算数组元素的累加和。具体使用方法如下：

import numpy as np

# 定义一个数组
arr = np.array([1, 2, 3, 4, 5])

# 计算数组元素的累加和
cumulative_sum = np.cumsum(arr)

# 输出结果
print(cumulative_sum)

运行结果如下：

[ 1  3  6 10 15]

可以看到，cumsum函数将数组元素逐个累加，得到了一个新的数组，新数组中的每个元素都是原数组中前面元素的累加和。