numpy.cumsum方法介绍
numpy.cumsum
是 NumPy 中的一个方法,用于计算数组元素的累积和(cumulative sum)。它沿着指定轴对元素进行逐个累加,生成一个新的数组,表示每个位置的累加结果。
语法:
numpy.cumsum(a, axis=None, dtype=None, out=None)
参数说明:
a
:输入数组。axis
:累加的轴,默认是沿平坦数组(即把数组展平成一维数组)进行累加。如果指定轴,则沿该轴进行累加。dtype
:累加时使用的数据类型,默认使用输入数组的类型。如果指定了dtype
,则会将累加的结果转换为该类型。out
:用于存储输出结果的数组,必须与输入数组形状相同。
返回值:
返回一个与输入数组 a
形状相同的数组,包含累积和的结果。
示例代码:
1. 一维数组的累积和:
import numpy as npa = np.array([1, 2, 3, 4])
result = np.cumsum(a)
print(result) # [ 1 3 6 10]
在这个例子中