学习NumPy中的cumsum()函数

NumPy中的cumsum()函数允许你计算沿某一轴的元素的累积和。

让我们来探讨一下。

函数语法

该函数的语法如下所示。

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

函数参数

该函数返回的参数如图所示。

1. a - 指的是输入数组。
2. axis - 沿哪个轴进行累积和。
3. dtype - 指定输出的数据类型。
4. out - 指定用于存储结果的输出数组。

函数的返回值

该函数返回一个新的数组，包含输入数组元素的累积和。

例子 #1

下面的代码显示了如何计算一个二维数组沿无轴的累积和。

# import numpy
import numpy as np
arr = np.array([[1,2,3], [4,5,6]])
print(f"result: {np.cumsum(arr, axis=None)}")

上面的代码应该将数组和一个保存元素累积和的数组平移。

一个输出的例子如图所示。

result: [ 1  3  6 10 15 21]

例子 #2

下面的例子展示了如何沿零轴使用cumsum()函数。

arr = np.array([[1,2,3], [4,5,6]])
print(f"result: {np.cumsum(arr, axis=0)}")

这应该返回

result:
[[1 2 3]
 [5 7 9]]

例子 #3

沿着 axis=1，函数返回的结果为。

arr = np.array([[1,2,3], [4,5,6]])
print(f"result: {np.cumsum(arr, axis=1)}")

输出的数组如图所示。

result:
[[ 1  3  6]
[ 4  9 15]]

结论

通过这篇文章，你学会了如何使用cumsum()函数来计算输入数组中沿给定轴的元素的累积和。请自由探索文档中的更多内容。

谢谢你的阅读！!