Python numpy.cumsum()

numpy.cumsum()函数用于计算给定轴上数组元素的累积和。

语法 : numpy.cumsum(arr, axis=None, dtype=None, out=None)

参数 :
arr :[array_like] 包含数字的数组，其累计之和是需要的。如果arr不是一个数组，将尝试进行转换。
axis :计算累积总和的坐标轴。默认是计算扁平化阵列的总和。
dtype :返回数组的类型，以及元素相乘的累加器的类型。如果没有指定dtype，默认为arr的dtype，除非arr的整数dtype的精度小于默认平台的整数。在这种情况下，将使用默认的平台整数来代替。
out :[ndarray, optional] 一个储存结果的位置。
-> 如果提供，它必须有一个输入广播到的形状。
-> 如果没有提供或没有，将返回一个新分配的数组。

返回 :除非指定了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]]