numpy.nancumsum

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

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

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

Введено в версии 1.12.0.

Параметры:

a : array_like Входной массив. axis : int, optional Ось, по которой вычисляется кумулятивная сумма. По умолчанию (None) вычисляется кумулятивная сумма по уплощенному массиву. dtype : dtype, optional Тип возвращаемого массива и аккумулятора, в котором суммируются элементы. Если dtype не указан, он по умолчанию равен типу данных a, за исключением случаев, когда a имеет целочисленный тип данных с точностью меньше, чем у целочисленного типа данных по умолчанию для платформы. В этом случае используется целочисленный тип данных по умолчанию для платформы. out : ndarray, optional Альтернативный выходной массив, в котором следует разместить результат. Он должен иметь ту же форму и длину буфера, что и ожидаемый выход, но тип будет преобразован при необходимости. Дополнительные сведения см. в doc.ufuncs (раздел «Аргументы вывода»).

Возвращает:

nancumsum : ndarray. Возвращается новый массив, содержащий результат, если не указан 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.]])