numpy.cumsum() en Python

numpy.cumsum()La función se usa cuando queremos calcular la suma acumulada de elementos de array sobre un eje dado.

Sintaxis: numpy.cumsum(arr, axis=Ninguno, dtype=Ninguno, out=Ninguno)

Parámetros:
arr: [array_like] Array que contiene números cuya suma acumulativa se desea. Si arr no es una array, se intenta una conversión.
eje : Eje a lo largo del cual se calcula la suma acumulada. El valor predeterminado es calcular la suma de la array aplanada.
dtype : Tipo del arreglo devuelto, así como del acumulador en el que se multiplican los elementos. Si no se especifica dtype, el valor predeterminado es el dtype de arr, a menos que arr tenga un dtype entero con una precisión menor que la del entero predeterminado de la plataforma. En ese caso, se utiliza en su lugar el entero de plataforma predeterminado.
out : [ndarray, opcional] Una ubicación en la que se almacena el resultado.
  -> Si se proporciona, debe tener una forma a la que se transmitan las entradas.
  -> Si no se proporciona o Ninguno, se devuelve una array recién asignada.

Devolución: se devuelve una nueva array que contiene el resultado, a menos que se especifique out, en cuyo caso se devuelve.

Código #1: Trabajando

# 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) 

Producción :

Input  number :  10
cumulative sum of input number :  [10]

 
Código #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) 

Producción :

Input array :  [[2 4 6]
 [1 3 5]]
cumulative sum of array elements:  [ 2  6 12 13 16 21]

 
Código #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) 

Producción :

Input array :  [[2 4 6]
 [1 3 5]]
cumulative sum of array elements taking axis 1:  [[ 2  6 12]
 [ 1  4  9]]