numpy.nancumsum

numpy.nancumsum(a, axis=None, dtype=None, out=None) [source]

Возвращает кумулятивную сумму элементов массива вдоль заданной оси, рассматривая значения Not a Number (NaN) как нули. Кумулятивная сумма не изменяется при встрече NaN, а ведущие NaN заменяются нулями.

Для слайсов, состоящих только из NaN или пустых, возвращаются нули.

Добавлена в версии 1.12.0.

Параметры

aarray_like

Входной массив.

axisint, optional

Ось, вдоль которой вычисляется кумулятивная сумма. По умолчанию (None) вычисляется кумулятивная сумма по уплощенному массиву.

dtypedtype, optional

Тип возвращаемого массива и аккумулятора, в котором суммируются элементы. Если dtype не указан, он по умолчанию равен типу a, за исключением случаев, когда у a есть целочисленный тип с точностью меньше, чем у стандартного целочисленного типа платформы. В этом случае используется стандартный целочисленный тип платформы.

outndarray, optional

Альтернативный выходной массив, в который помещается результат. Он должен иметь ту же форму и длину буфера, что и ожидаемый результат, но тип может быть преобразован при необходимости. Смотрите ufuncs-output-type для получения дополнительной информации.

Возвращает

nancumsumndarray.

Возвращается новый массив, содержащий результат, если out не указан, в противном случае возвращается он. Результат имеет тот же размер, что и a, и ту же форму, что и a, если axis не равен None или a является одномерным массивом.

См. также

numpy.cumsum

Кумулятивная сумма по массиву, обрабатывающая NaN.

isnan

Показывает, какие элементы являются NaN.

Примеры

>>> np.nancumsum(1)
array([1])
>>> np.nancumsum([1])
array([1])
>>> np.nancumsum([1, np.nan])
array([1.,  1.])
>>> a = np.array([[1, 2], [3, np.nan]])
>>> np.nancumsum(a)
array([1.,  3.,  6.,  6.])
>>> np.nancumsum(a, axis=0)
array([[1.,  2.],
       [4.,  2.]])
>>> np.nancumsum(a, axis=1)
array([[1.,  3.],
       [3.,  3.]])