Somas NumPy


Somas

Qual é a diferença entre soma e adição?

A adição é feita entre dois argumentos, enquanto a soma ocorre sobre n elementos.

Exemplo

Adicione os valores em arr1 aos valores em arr2:

import numpy as np

arr1 = np.array([1, 2, 3])
arr2 = np.array([1, 2, 3])

newarr = np.add(arr1, arr2)

print(newarr)

Devoluções: [2 4 6]

Exemplo

Some os valores em arr1 e os valores em arr2:

import numpy as np

arr1 = np.array([1, 2, 3])
arr2 = np.array([1, 2, 3])

newarr = np.sum([arr1, arr2])

print(newarr)

Devoluções: 12


Soma sobre um eixo

Se você especificar axis=1, NumPy somará os números em cada array.

Exemplo

Execute a soma na seguinte matriz sobre o 1º eixo:

import numpy as np

arr1 = np.array([1, 2, 3])
arr2 = np.array([1, 2, 3])

newarr = np.sum([arr1, arr2], axis=1)

print(newarr)

Devoluções: [6 6]


Soma cumulativa

Soma cumulativa significa adicionar parcialmente os elementos na matriz.

Por exemplo, a soma parcial de [1, 2, 3, 4] seria [1, 1+2, 1+2+3, 1+2+3+4] = [1, 3, 6, 10].

Faça a soma parcial com a cumsum()função.

Exemplo

Execute a soma cumulativa na seguinte matriz:

import numpy as np

arr = np.array([1, 2, 3])

newarr = np.cumsum(arr)

print(newarr)

Devoluções: [1 3 6]