NumPy参考 >例行程序 >Mathematical functions > numpy.cumsum
numpy.
cumsum
(a,axis = None,dtype = None,out = None )[源代码] ¶返回沿给定轴的元素的累加和。
输入数组。
计算累计和的轴。默认值(无)是计算展平数组上的总和。
返回的数组和累加器的类型,元素在其中累加。如果dtype
没有指定,则默认为的D型一,除非一个具有精度小于默认平台整数的整数D型。在这种情况下,将使用默认平台整数。
放置结果的替代输出数组。它必须具有与预期输出相同的形状和缓冲区长度,但是如果需要,将强制转换类型。有关更多详细信息,请参见ufuncs-output-type。
除非指定out,否则将返回保存结果的新数组,在这种情况下,将返回对out的引用。结果有大小相同,一,且相同的形状一如果 轴不是无或一个有1 d阵列。
笔记
使用整数类型时,算术是模块化的,并且在溢出时不会引发错误。
例子
>>> 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]])