Рубрики
Без рубрики

Numpy – сумма элементов в массиве – сумма ()

Автор оригинала: Python Examples.

Numpy Sum ()

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

В этом руководстве мы узнаем, как использовать функцию SUM () в наших программах Python.

Синтаксис – numpy.sum ()

Синтаксис Numpy.sum () показан ниже.

numpy.sum(a, axis=None, dtype=None, out=None, keepdims=, initial=)

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

Пример 1: numpy sum ()

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

Python Program

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: Numpy Sum () вдоль оси

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

Python Program

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 Отказ

Python Program

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. Но, но если вы укажете начальное значение, сумма будет начальное значение + сумма (массив) вдоль оси или всего, что согласно аргументам.

Python Program

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   

Резюме

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