Veamos cómo calcular la suma de todas las columnas de una array NumPy bidimensional.
Ejemplo :
Input : [[1, 2, 3, 4, 5], [5, 6, 7, 8, 9], [2, 1, 5, 7, 8], [2, 9, 3, 1, 0]] Output : [10, 18, 18, 20, 22] Input : [[5, 4, 1, 7], [0, 9, 3, 5], [3, 2, 8, 6]] Output : [8, 15, 12, 18]
Enfoque 1: Usaremos el sum()
método. Pasaremos el parámetro axis = 0
para obtener las columnas de suma sabias.
# importing numpy import numpy as np # initialize the 2-d array arr = np.array([[1, 2, 3, 4, 5], [5, 6, 7, 8, 9], [2, 1, 5, 7, 8], [2, 9, 3, 1, 0]]) # calculating column wise sum sum_2d = arr.sum(axis = 0) # displaying the sum print("Column wise sum is :\n", sum_2d)
Producción :
Column wise sum is : [10 18 18 20 22]
Enfoque 2: También podemos usar el numpy.einsum()
método, con el parámetro 'ij->j'
.
# importing numpy import numpy as np # initialize the 2-d array arr = np.array([[1, 2, 3, 4, 5], [5, 6, 7, 8, 9], [2, 1, 5, 7, 8], [2, 9, 3, 1, 0]]) # calculating column wise sum sum_2d = np.einsum('ij->j', arr) # displaying the sum print("Column wise sum is :\n", sum_2d)
Producción :
Column wise sum is : [10 18 18 20 22]
Publicación traducida automáticamente
Artículo escrito por aashishsaxena11222 y traducido por Barcelona Geeks. The original can be accessed here. Licence: CCBY-SA