np.cumsum()
时间: 2024-06-16 20:06:18 浏览: 131
指定轴-stm32 ll库使用指南+新建工程模板方法
np.cumsum()是NumPy库中的一个函数,用于计算数组元素的累积和。它返回一个与输入数组相同大小的新数组,其中每个元素是原始数组中该位置及之前所有元素的和。
以下是np.cumsum()的一些特点:
1. 如果输入数组是多维的,np.cumsum()默认按照行方向进行计算。
2. 可以通过指定axis参数来改变计算方向。
3. np.cumsum()可以应用于不同的数据类型,包括整数、浮点数和复数。
下面是一个示例,展示了np.cumsum()的使用方法:
```python
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]
```
阅读全文