numpy.cumsum()函数用于计算给定轴上数组元素的累积和。
用法: numpy.cumsum(arr, axis=None, dtype=None, out=None)
参数:
arr :[数组]包含需要累积总和的数字的数组。如果arr不是数组,则尝试进行转换。
axis :计算累计和的轴。默认值是计算展平数组的总和。
dtype :返回数组的类型,以及与元素相乘的累加器的类型。如果未指定dtype,则默认为arr的dtype,除非arr的整数dtype的精度小于默认平台整数的精度。在这种情况下,将使用默认平台整数。
out :[ndarray,可选]将结果存储到的位置。
->如果提供,则必须具有广播输入的形状。
->如果未提供或没有,则返回新分配的数组。
Return :除非指定out,否则将返回保存结果的新数组,在这种情况下将返回该数组。
代码1:工作
# Python program explaining
# numpy.cumsum() function
import numpy as geek
in_num = 10
print ("Input number:", in_num)
out_sum = geek.cumsum(in_num)
print ("cumulative sum of input number:", out_sum)
输出:
Input number: 10 cumulative sum of input number: [10]
代码2:
# Python program explaining
# numpy.cumsum() function
import numpy as geek
in_arr = geek.array([[2, 4, 6], [1, 3, 5]])
print ("Input array:", in_arr)
out_sum = geek.cumsum(in_arr)
print ("cumulative sum of array elements:", out_sum)
输出:
Input array: [[2 4 6] [1 3 5]] cumulative sum of array elements: [ 2 6 12 13 16 21]
代码3:
# Python program explaining
# numpy.cumsum() function
import numpy as geek
in_arr = geek.array([[2, 4, 6], [1, 3, 5]])
print ("Input array:", in_arr)
out_sum = geek.cumsum(in_arr, axis = 1)
print ("cumulative sum of array elements taking axis 1:", out_sum)
输出:
Input array: [[2 4 6] [1 3 5]] cumulative sum of array elements taking axis 1: [[ 2 6 12] [ 1 4 9]]
相关用法
注:本文由纯净天空筛选整理自jana_sayantan大神的英文原创作品 numpy.cumsum() in Python。非经特殊声明,原始代码版权归原作者所有,本译文未经允许或授权,请勿转载或复制。