Soma NumPy() – Acervo Lima Tutorials

Resumo : neste tutorial, você aprenderá como usar a sum()função numpy para retornar a soma de todos os elementos em um array.

Introdução à função numpy sum()

A função numpy sum()é uma função agregada que pega um array e retorna a soma de todos os elementos.

O exemplo a seguir usa a sum()função para calcular a soma de todos os elementos de uma matriz 1-D:

import numpy as np

a = np.array([1, 2, 3])
total = np.sum(a)
print(total)Linguagem de código:  Python  ( python )

Saída:

6Linguagem de código:  Python  ( python )

Como funciona.

Primeiro, crie um novo array numpy que tenha três números 1, 2 e 3:

a = np.array([1, 2, 3])Linguagem de código:  Python  ( python )

Segundo, calcule a soma de todos os elementos do array a usando a sum()função:

total = np.sum(a)Linguagem de código:  Python  ( python )

Terceiro, exiba o resultado:

print(total)Linguagem de código:  Python  ( python )

O exemplo a seguir usa a sum()função para calcular a soma de todos os elementos de uma matriz 2D:

import numpy as np

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

total = np.sum(a)
print(total)Linguagem de código:  Python  ( python )

Saída:

21Linguagem de código:  Python  ( python )

Neste exemplo, sum()soma todos os números da matriz a.

A sum()função também aceita o argumento do eixo que permite retornar a soma dos elementos de um eixo. Por exemplo:

import numpy as np

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

total = np.sum(a, axis=0)
print(total)Linguagem de código:  Python  ( python )

Saída:

[5 7 9]Linguagem de código:  Python  ( python )

Neste exemplo, a sum()função retorna um novo array onde cada elemento é a soma dos elementos do array ano eixo 0.

Da mesma forma, você pode somar os elementos no eixo 1 assim:

import numpy as np

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

total = np.sum(a, axis=1)
print(total)Linguagem de código:  Python  ( python )

Saída:

[ 6 15]Linguagem de código:  Python  ( python )

Cópia NumPy()