Сумма элементов массива Numpy sum() в Python

Чтобы получить сумму всех элементов в массиве numpy, вы можете использовать встроенную функцию Numpy sum() в Python.

Синтаксис

Синтаксис функции показан ниже.

numpy.sum(a, axis=None, dtype=None, out=None, keepdims=<no value>, initial=<no value>)

Мы разберемся с параметрами в определении функции, используя нижеприведенные примеры.

Пример 1

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

import numpy as np

a = np.array([4, 5, 3, 7])
print('input\n',a)

b = np.sum(a)
print('sum\n',b)

Вывод:

input
 [4 5 3 7]
sum
 19

Объяснение:

4 + 5 + 3 + 7 = 19

Пример 2: по оси

Вы можете указать ось для sum() и, таким образом, получить сумму элементов вдоль оси.

import numpy as np

a = np.array([4, 5, 3, 7]).reshape(2, 2)
print('input\n',a)

b = np.sum(a, axis=0)
print('sum\n',b)

Вывод:

input
 [[4 5]
 [3 7]]
sum
 [ 7 12]

Объяснение:

 [[4  5]
   +  +
  [3  7]]
------------
  [7  12]

В приведенной выше программе мы нашли сумму по оси = 0. Теперь давайте попробуем с осью = 1.

import numpy as np

a = np.array([4, 5, 3, 7]).reshape(2, 2)
print('input\n',a)

b = np.sum(a, axis=1)
print('sum\n',b)

Вывод:

input
 [[4 5]
 [3 7]]
sum
 [ 9 10]

Объяснение:

[4 + 5] = 9
[3 + 7] = 10

Hence [9 10]

Пример 3: начальное значение суммы

Вы также можете указать начальное значение суммы.

По умолчанию начальное значение равно 0. Но, если вы укажете начальное значение, сумма будет равна начальному значению + сумма (массив) по оси, согласно аргументам.

import numpy as np

a = np.array([4, 5, 3, 7])
print('input\n',a)

b = np.sum(a, initial=52)
print('sum\n',b)

Вывод:

input
 [4 5 3 7]
sum
 71

Объяснение:

sum(a, initial=52) = sum(a) + initial
                   = sum([4 5 3 7]) + 52
                   = 19 + 52
                   = 71

В этом учебном пособии по примерам Python мы узнали, как получить сумму элементов в массиве numpy или по оси с помощью numpy.sum().