原文:https://docs.scipy.org/doc/numpy/reference/generated/numpy.cumsum.html
校对:(虚位以待)
numpy.
cumsum
(a, axis=None, dtype=None, out=None)[source]返回沿给定轴的元素的累积和。
参数: | a:array_like
axis:int,可选
dtype:dtype,可选
out:ndarray,可选
|
---|---|
返回: | cumsum_along_axis:ndarray。
|
笔记
当使用整数类型时,算术是模块化的,并且在溢出时不产生错误。
例子
>>> a = np.array([[1,2,3], [4,5,6]])
>>> a
array([[1, 2, 3],
[4, 5, 6]])
>>> np.cumsum(a)
array([ 1, 3, 6, 10, 15, 21])
>>> np.cumsum(a, dtype=float) # specifies type of output value(s)
array([ 1., 3., 6., 10., 15., 21.])
>>> np.cumsum(a,axis=0) # sum over rows for each of the 3 columns
array([[1, 2, 3],
[5, 7, 9]])
>>> np.cumsum(a,axis=1) # sum over columns for each of the 2 rows
array([[ 1, 3, 6],
[ 4, 9, 15]])