Функция NumPy sum() в Python: сумма элементов массива

Знакомство с функцией numpy sum()

Функция numpy sum() — это агрегатная функция в Python, которая принимает массив и возвращает сумму всех элементов.

Пример 1.

В данном примере функция sum() используется для вычисления суммы всех элементов одномерного массива:

import numpy as np

a = np.array([1, 2, 3])
total = np.sum(a)
print(total)

Выход:

6

Как это работает.

• Сначала создайте новый массив numpy, содержащий три числа 1, 2 и 3:

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

• Во-вторых, вычислите сумму всех элементов массива a с помощью функции sum():

total = np.sum(a)

• В-третьих, отобразите результат:

print(total)

Пример 2.

В следующем примере функция sum() используется для вычисления суммы всех элементов двумерного массива:

import numpy as np

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

total = np.sum(a)
print(total)

Выход:

21

В этом примере функция sum() складывает все числа массива a.

Функция sum() также принимает аргумент оси, который позволяет вам возвращать сумму элементов оси. Например:

import numpy as np

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

total = np.sum(a, axis=0)
print(total)

Выход:

[5 7 9]

В этом примере функция sum() возвращает новый массив, где каждый элемент представляет собой сумму элементов массива a на оси-0.

Аналогичным образом вы можете суммировать элементы по оси-1 следующим образом:

import numpy as np

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

total = np.sum(a, axis=1)
print(total)

Выход:

[ 6 15]