numpy.sum() em Python

A função `numpy sum()` do Python é usada para obter a soma dos elementos de um array ao longo de um eixo dado.

A sintaxe da função `numpy sum()` do Python é:

A sintaxe do método `sum()` do NumPy em Python é:

sum(array, axis, dtype, out, keepdims, initial)

• Os elementos do array são usados para calcular a soma.
• Se o eixo não for fornecido, a soma de todos os elementos é retornada. Se o eixo for uma tupla de inteiros, a soma de todos os elementos nos eixos dados é retornada.
• Podemos especificar dtype para especificar o tipo de dados de saída retornado.
• A variável out é usada para especificar o array onde colocar o resultado. É um parâmetro opcional.
• O parâmetro keepdims é um parâmetro booleano. Se for definido como True, os eixos que são reduzidos são deixados no resultado como dimensões com tamanho um.
• O parâmetro initial especifica o valor inicial para a soma.

Exemplos de `numpy sum()` em Python

Vamos ver alguns exemplos da função `numpy sum()`.

1. Soma de Todos os Elementos no Array

Se passarmos apenas o array na função sum(), ele é achatado e a soma de todos os elementos é retornada.

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

Saída: Soma de todos os elementos é 21

2. Soma dos Elementos do Array Ao Longo do Eixo

Se especificarmos o valor do eixo, a soma dos elementos ao longo desse eixo é retornada. Se o formato do array é (X, Y), então a soma ao longo do eixo 0 será de formato (1, Y). A soma ao longo do eixo 1 será de formato (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}')

Saída:

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

3. Especificando o Tipo de Dado de Saída da Soma

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

Saída: Soma dos elementos no eixo 1 é [3. 7.]

4. Valor Inicial para a Soma

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

Saída: A soma dos elementos no eixo 1 é [13 17] Referência: Documento da API