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