numpy.sum

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

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

Параметры:

a : array_like Элементы для суммирования. axis : None or int or tuple of ints, optional Ось или оси, по которым выполняется суммирование. По умолчанию, axis=None, суммируются все элементы входного массива. Если axis отрицательный, он отсчитывается с последней до первой оси. Добавлен в версии 1.7.0. Если axis является кортежем целых чисел, суммирование выполняется по всем осям, указанным в кортеже, вместо одной оси или всех осей, как раньше. dtype : dtype, optional Тип возвращаемого массива и аккумулятора, в котором суммируются элементы. Тип dtype a используется по умолчанию, если a не имеет целочисленного типа с меньшей точностью, чем по умолчанию. В этом случае, если a со знаком, используется целочисленный тип платформы, а если a без знака, используется беззнаковый целочисленный тип той же точности, что и целочисленный тип платформы. out : ndarray, optional Альтернативный массив вывода, в который нужно поместить результат. Он должен иметь такую же форму, как ожидаемый результат, но тип значений вывода будет преобразован при необходимости. keepdims : bool, optional Если это установлено в True, оси, которые уменьшаются, остаются в результате в качестве измерений с размером один. С этим вариантом результат будет правильно транслироваться относительно входного массива. Если передано значение по умолчанию, то keepdims не будет передано в метод sum подклассов ndarray, однако любое отличное от значения по умолчанию значение будет. Если метод подкласса не реализует keepdims, будут подняты исключения. initial : scalar, optional Начальное значение для суммы. Подробности см. в reduce. Добавлен в версии 1.15.0.

Возвращает:

sum_along_axis : ndarray Массив с такой же формой, как a, с удаленной заданной осью. Если a является массивом размерности 0, или если axis равно None, возвращается скаляр. Если указан массив вывода, возвращается ссылка на out.

См. также

ndarray.sum

Эквивалентный метод.

cumsum

Кумулятивная сумма элементов массива.

trapz

Интегрирование значений массива с помощью составного правила трапеций.

mean, average

Примечания

Арифметика является модульной при использовании целочисленных типов и не вызывает ошибок при переполнении.

Сумма пустого массива равна нейтральному элементу 0:

>>> np.sum([])
0.0

Примеры

>>> np.sum([0.5, 1.5])
2.0
>>> np.sum([0.5, 0.7, 0.2, 1.5], dtype=np.int32)
1
>>> np.sum([[0, 1], [0, 5]])
6
>>> np.sum([[0, 1], [0, 5]], axis=0)
array([0, 6])
>>> np.sum([[0, 1], [0, 5]], axis=1)
array([1, 5])

Если аккумулятор слишком мал, происходит переполнение:

>>> np.ones(128, dtype=np.int8).sum(dtype=np.int8)
-128

Вы также можете начать суммирование со значения, отличного от нуля:

>>> np.sum([10], initial=5)
15