Calcule la suma de todas las columnas en una array 2D NumPy

Veamos cómo calcular la suma de todas las columnas en una array 2D NumPy.
Método 1: usar un bucle anidado para acceder a los elementos de la array en forma de columna y luego almacenar su suma en una variable y luego imprimirla.
Ejemplo 1: 
 

Python3

# importing required libraries
import numpy
 
# explicit function to compute column wise sum
def colsum(arr, n, m):
    for i in range(n):
        su = 0;
        for j in range(m):
            su += arr[j][i]
        print(su, end = " ")   
 
# creating the 2D Array
TwoDList = [[1, 2, 3], [4, 5, 6],
            [7, 8, 9], [10, 11, 12]]
TwoDArray = numpy.array(TwoDList)
 
# displaying the 2D Array
print("2D Array:")
print(TwoDArray)
 
# printing the sum of each column
print("\nColumn-wise Sum:")
colsum(TwoDArray, len(TwoDArray[0]), len(TwoDArray))

Producción : 
 

2D Array:
[[ 1  2  3]
 [ 4  5  6]
 [ 7  8  9]
 [10 11 12]]

Column-wise Sum:
22 26 30 

Ejemplo 2: 
 

Python3

# importing required libraries
import numpy
 
# explicit function to compute column wise sum
def colsum(arr, n, m):
    for i in range(n):
        su = 0;
        for j in range(m):
            su += arr[j][i]
        print(su, end = " ")   
 
# creating the 2D Array
TwoDList = [[1.2, 2.3], [3.4, 4.5]]
TwoDArray = numpy.array(TwoDList)
 
# displaying the 2D Array
print("2D Array:")
print(TwoDArray)
 
# printing the sum of each column
print("\nColumn-wise Sum:")
colsum(TwoDArray, len(TwoDArray[0]), len(TwoDArray))

Producción : 
 

2D Array:
[[1.2 2.3]
 [3.4 4.5]]

Column-wise Sum:
4.6 6.8 

Método 2: Usando la función sum() en NumPy, la función numpy.sum(arr, axis, dtype, out) devuelve la suma de los elementos de la array sobre el eje especificado. Para calcular la suma de todas las columnas, el argumento del eje debe ser 0 en la función sum().
Ejemplo 1 : 
 

Python3

# importing required libraries
import numpy
 
# creating the 2D Array
TwoDList = [[1, 2, 3], [4, 5, 6],
            [7, 8, 9], [10, 11, 12]]
TwoDArray = numpy.array(TwoDList)
 
# displaying the 2D Array
print("2D Array:")
print(TwoDArray)
 
# printing the sum of each column
print("\nColumn-wise Sum:")
print(numpy.sum(TwoDArray, axis = 0))

Producción : 
 

2D Array:
[[ 1  2  3]
 [ 4  5  6]
 [ 7  8  9]
 [10 11 12]]

Column-wise Sum:
22 26 30

Ejemplo 2: 
 

Python3

# importing required libraries
import numpy
 
# creating the 2D Array
TwoDList =[[1.2, 2.3], [3.4, 4.5]]
TwoDArray = numpy.array(TwoDList)
 
# displaying the 2D Array
print("2D Array:")
print(TwoDArray)
 
# printing the sum of each column
print("\nColumn-wise Sum:")
print(*numpy.sum(TwoDArray, axis = 0))

Producción : 
 

2D Array:
[[1.2 2.3]
 [3.4 4.5]]

Column-wise Sum:
4.6 6.8

Publicación traducida automáticamente

Artículo escrito por riturajsaha y traducido por Barcelona Geeks. The original can be accessed here. Licence: CCBY-SA

Deja una respuesta

Tu dirección de correo electrónico no será publicada. Los campos obligatorios están marcados con *