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

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

Возвращает

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.]])