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, суммируются все элементы входного массива. Если ось отрицательна, она отсчитывается с последней до первой оси. Введено в версии 1.7.0. Если ось является кортежем целых чисел, суммирование выполняется по всем осям, указанным в кортеже, а не по одной оси или всем осям, как раньше. dtype : dtype, optional Тип возвращаемого массива и аккумулятора, в котором суммируются элементы. Тип данных 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