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