numpy.sum () в Python - pythobyte.com

Функция Python Numpy Sum () используется для получения суммы элементов массива на заданную ось.

Python Numpy Sum () Синтаксис функции

Синтаксис метода Sumpy () Python Numpy ():

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

массив Элементы используются для расчета суммы.
Если Ось Не предусмотрено, сумма всех элементов возвращается. Если ось представляет собой кортеж INT, сумма всех элементов в данных осях возвращается.
Мы можем указать Dtype Чтобы указать возвращенный тип данных.
OUT Переменная используется для указания массива для размещения результата. Это дополнительный параметр.
MeadDims это логический параметр. Если это установлено значение true, оси, которые уменьшены, остаются в результате как размеры с размером один.
Первоначальный Параметр указывает исходное значение для суммы.

Примеры Numpy () Python Numpy ()

Давайте посмотрим на некоторые примеры функции numpy sum ().

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. Сумма элементов массива вдоль оси

Если мы укажем значение оси, возвращается сумма элементов вдоль этой оси. Если форма массива is (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 док