El método numpy.sum() “calcula la suma de todos los elementos en una matriz NumPy” o a lo largo de un eje específico.
Sintaxis del método numpy.sum()
numpy.sum(array, axis=None, dtype=None, out=None)
Parámetros
- Una array es una matriz que se va a sumar.
- Un axis es un eje a lo largo del cual se suman los elementos. Si el eje es Ninguno, los detalles de la matriz se suman.
- El dtype es el tipo de datos de la matriz resultante. Si dtype es Ninguno, el tipo de datos de la matriz resultante es el mismo que el tipo de datos de la matriz de entrada.
- El out es una matriz de salida. Si out no es none el resultado se almacena en out.
Ejemplo
import numpy as np
# Sample NumPy array
arr = np.array([[1, 2, 3], [4, 5, 6]])
print("Array:")
print(arr)
# Compute the sum of all elements
total_sum = np.sum(arr)
print("\nSum of all elements:", total_sum)
# Compute the sum along axis 0 (rows)
row_sum = np.sum(arr, axis=0)
print("\nSum along axis 0 (rows):", row_sum)
# Compute the sum along axis 1 (columns)
col_sum = np.sum(arr, axis=1)
print("\nSum along axis 1 (columns):", col_sum)
Producción
Array:
[[1 2 3]
[4 5 6]]
Sum of all elements: 21
Sum along axis 0 (rows): [5 7 9]
Sum along axis 1 (columns): [ 6 15]
En este código, creamos una matriz NumPy de muestra.
En el siguiente paso, calculamos la suma de todos los elementos, la suma a lo largo del eje 0 (filas) y la suma a lo largo del eje 1 (columnas) usando la función numpy.sum().
Las sumas resultantes se imprimen y muestran cómo se puede calcular la suma usando la función numpy.sum().