numpy.sum() в Python

Python numpy sum() функция используется для получения суммы элементов массива по заданной оси.

Синтаксис функции Python numpy sum()

Синтаксис метода sum() в библиотеке NumPy:

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

• Элементы массива используются для вычисления суммы.
• Если ось не указана, возвращается сумма всех элементов. Если ось представляет собой кортеж целых чисел, возвращается сумма элементов в указанных осях.
• Мы можем указать dtype, чтобы задать тип данных вывода.
• Переменная out используется для указания массива для размещения результата. Это необязательный параметр.
• Параметр keepdims является логическим. Если он установлен в True, оси, которые уменьшаются, остаются в результате как размерности с единичным размером.
• Параметр initial указывает начальное значение для суммы.

Примеры использования функции Python numpy sum()

Давайте рассмотрим несколько примеров использования функции sum() в библиотеке numpy.

1. Сумма всех элементов в массиве

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

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

Выходные данные: Сумма всех элементов равна 21

2. Сумма элементов массива вдоль оси

Если мы указываем значение оси, возвращается сумма элементов вдоль этой оси. Если форма массива (X, Y), то сумма вдоль оси 0 будет иметь форму (1, Y). Сумма вдоль оси 1 будет иметь форму (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}')

Выходные данные:

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

3. Указание типа данных для суммы

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

Выходные данные: Сумма элементов по оси 1 составляет [3. 7.]

4. Начальное значение для суммы

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

Вывод: Сумма элементов по оси 1 равна [13 17] Ссылка: Документация по API