np.array.cumsum()
时间: 2023-11-20 19:17:10 浏览: 68
python numpy中cumsum的用法详解
5星 · 资源好评率100%
`np.array.cumsum()`是NumPy库中的一个函数,用于计算数组中元素的累积和。它返回一个新的数组,其中每个元素都是原始数组中对应位置之前所有元素的和。
例如,假设我们有一个NumPy数组`arr`:
```python
import numpy as np
arr = np.array([1, 2, 3, 4, 5])
```
我们可以使用`np.array.cumsum()`计算累积和:
```python
cumulative_sum = np.array.cumsum(arr)
print(cumulative_sum)
```
输出:
```
[ 1 3 6 10 15]
```
在这个例子中,输出的数组元素分别是:1、1+2=3、1+2+3=6、1+2+3+4=10、1+2+3+4+5=15。
阅读全文