Python numpy nancumsum用法及代码示例

本文简要介绍 python 语言中 numpy.nancumsum 的用法。

用法:

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

返回给定轴上数组元素的累积和，将非数字 (NaN) 视为零。当遇到NaNs并且前导NaNs被零替换时，累积和不会改变。

对于 all-NaN 或空的切片返回零。

参数：

a： array_like

输入数组。

axis： 整数，可选

计算累积和的轴。默认值 (None) 是计算展平数组上的 cumsum。

dtype： dtype，可选

返回的数组以及对元素进行求和的累加器的类型。如果numpy.dtype未指定，默认为 dtypea， 除非a具有精度小于默认平台整数的整数数据类型。在这种情况下，将使用默认的平台整数。

out： ndarray，可选

用于放置结果的替代输出数组。它必须具有与预期输出相同的形状和缓冲区长度，但如果需要，类型将被强制转换。有关更多详细信息，请参阅输出类型确定。

返回：

nancumsum： 数组。

除非指定了 out，否则将返回一个保存结果的新数组，并在其中返回它。如果轴不是 None 或 a 是一维数组，则结果具有与 a 相同的大小和相同的形状。

例子：

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