np.sum()

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

Сигнатураnp.sum(a, axis=None)

Без axis суммирует все элементы. С axis сворачивает указанную ось.

import numpy as np
a = np.array([[1, 2], [3, 4]])
print(np.sum(a))          # 10
print(np.sum(a, axis=0))  # [4 6]