numpy.sum() en Python

La función sum() de numpy de Python se utiliza para obtener la suma de los elementos de una matriz sobre un eje dado.

Sintaxis de la función sum() de Python numpy

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 enteros, se devuelve la suma de todos los elementos en los ejes dados.
• Podemos especificar dtype para especificar el tipo de datos de salida devuelto.
• La variable out se utiliza para especificar la matriz en la que colocar el resultado. Es un parámetro opcional.
• El parámetro keepdims es un 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 initial especifica el valor inicial para la suma.

Ejemplos de la función sum() de numpy en Python

Veamos algunos ejemplos de la función sum() de numpy.

1. Suma de Todos los Elementos en el Array

Si solo pasamos el array en la función sum(), este se aplana y se devuelve la suma de todos los elementos.

import numpy as np

array1 = np.array(
    [[1, 2],
     [3, 4],
     [5, 6]])

total = np.sum(array1)
print(f'Sum of all the elements is {total}')

Salida: La suma de todos los elementos es 21

2. Suma de Elementos del Array 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 del array es (X, Y), entonces la suma a lo largo del eje 0 tendrá forma (1, Y). La suma a lo largo del eje 1 tendrá forma (1, X).

import numpy as np

array1 = 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}')

Salida:

Sum of elements at 0-axis is [ 9 12]
Sum of elements at 1-axis is [ 3  7 11]

3. Especificando el Tipo de Dato de Salida de la Suma

import numpy as np

array1 = 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}')

Salida: La suma de los elementos en el eje 1 es [3. 7.]

4. Valor Inicial para la Suma

import numpy as np

array1 = 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: La suma de los elementos en el eje 1 es [13 17] Referencia: Documentación de la API