当前位置: 首页>>代码示例 >>用法及示例精选 >>正文


Python numpy.cumsum()用法及代码示例


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。非经特殊声明,原始代码版权归原作者所有,本译文未经允许或授权,请勿转载或复制。