La función sum() de numpy de Python se utiliza para obtener la suma de los elementos de una matriz sobre un eje determinado.
Sintaxis de la función sum() de numpy en Python
La sintaxis del método sum() de Python NumPy es:
sum(array, axis, dtype, out, keepdims, initial)- Los elementos de la matriz se utilizan para calcular la suma.
- Si no se proporciona el eje , se devuelve la suma de todos los elementos. Si el eje es una tupla de números enteros, se devuelve la suma de todos los elementos de los ejes indicados.
- Podemos especificar dtype para especificar el tipo de datos de salida devueltos.
- La variable out se utiliza para especificar la matriz donde se colocará el resultado. Es un parámetro opcional.
- Keepdims es un parámetro booleano . Si se establece en True, los ejes que se reducen se dejan en el resultado como dimensiones con tamaño uno.
- El parámetro inicial especifica el valor inicial para la suma.
Ejemplos de suma numpy en Python
Veamos algunos ejemplos de la función sum() de numpy.
1. Suma de todos los elementos de la matriz
Si pasamos sólo la matriz en la función sum(), se aplana y se devuelve la suma de todos los elementos.
import numpy as nparray1 = np.array( [[1, 2], [3, 4], [5, 6]])total = np.sum(array1)print(f'Sum of all the elements is {total}')Producción :Sum of all the elements is 21
2. Suma de elementos de matriz a lo largo del eje
Si especificamos el valor del eje, se devuelve la suma de los elementos a lo largo de ese eje. Si la forma de la matriz es (X, Y), la suma a lo largo del eje 0 tendrá la forma (1, Y). La suma a lo largo del eje 1 tendrá la forma (1, X).
import numpy as nparray1 = np.array( [[1, 2], [3, 4], [5, 6]])total_0_axis = np.sum(array1, axis=0)print(f'Sum of elements at 0-axis is {total_0_axis}')total_1_axis = np.sum(array1, axis=1)print(f'Sum of elements at 1-axis is {total_1_axis}')Producción:
Sum of elements at 0-axis is [ 9 12]Sum of elements at 1-axis is [ 3 7 11]3. Especificación del tipo de datos de salida de la suma
import numpy as nparray1 = np.array( [[1, 2], [3, 4]])total_1_axis = np.sum(array1, axis=1, dtype=float)print(f'Sum of elements at 1-axis is {total_1_axis}')Producción :Sum of elements at 1-axis is [3. 7.]
4. Valor inicial de la suma
import numpy as nparray1 = np.array( [[1, 2], [3, 4]])total_1_axis = np.sum(array1, axis=1, initial=10)print(f'Sum of elements at 1-axis is {total_1_axis}')Salida : Sum of elements at 1-axis is [13 17] Referencia : Documentación de API