Python numpy nansum用法及代碼示例

本文簡要介紹 python 語言中 numpy.nansum 的用法。

用法:

numpy.nansum(a, axis=None, dtype=None, out=None, keepdims=<no value>, initial=<no value>, where=<no value>)

返回給定軸上的數組元素的總和，將非數字 (NaN) 視為零。

在 NumPy 版本 <= 1.9.0 中，對於 all-NaN 或空的切片返回 Nan。在更高版本中返回零。

參數：

a： array_like

包含需要其總和的數字的數組。如果 a 不是數組，則嘗試轉換。

axis： {int，int 元組，無}，可選

計算總和的一個或多個軸。默認是計算展平數組的總和。

dtype： 數據類型，可選

返回數組的類型和元素相加的累加器的類型。默認情況下，使用 a 的 dtype。一個例外是當 a 的整數類型的精度低於平台 (u)intp 時。在這種情況下，默認值將是 (u)int32 或 (u)int64，具體取決於平台是 32 位還是 64 位。對於不精確的輸入，dtype 必須是不精確的。

out： ndarray，可選

用於放置結果的備用輸出數組。默認為 None 。如果提供，它必須具有與預期輸出相同的形狀，但如果需要，類型將被轉換。有關更多詳細信息，請參閱輸出類型確定。將 NaN 轉換為整數可能會產生意外結果。

keepdims： 布爾型，可選

如果將其設置為 True，則縮小的軸將作為尺寸為 1 的尺寸留在結果中。使用此選項，結果將針對原始 a 正確廣播。

如果該值不是默認值，則保持昏暗將被傳遞到numpy.mean或者numpy.sum子類的方法numpy.ndarray。如果子類方法沒有實現保持昏暗將引發任何異常。

initial： 標量，可選

總和的起始值。有關詳細信息，請參閱 reduce 。

where： 類似於 bool 的數組，可選

要包含在總和中的元素。有關詳細信息，請參閱 reduce 。

返回：

nansum： 數組。

除非指定了 out，否則將返回一個保存結果的新數組，並在其中返回它。如果軸不是 None 或 a 是一維數組，則結果具有與 a 相同的大小和相同的形狀。

注意：

如果同時存在正無窮大和負無窮大，則總和將不是數字 (NaN)。

例子：

>>> np.nansum(1)
1
>>> np.nansum([1])
1
>>> np.nansum([1, np.nan])
1.0
>>> a = np.array([[1, 1], [1, np.nan]])
>>> np.nansum(a)
3.0
>>> np.nansum(a, axis=0)
array([2.,  1.])
>>> np.nansum([1, np.nan, np.inf])
inf
>>> np.nansum([1, np.nan, np.NINF])
-inf
>>> from numpy.testing import suppress_warnings
>>> with suppress_warnings() as sup:
...     sup.filter(RuntimeWarning)
...     np.nansum([1, np.nan, np.inf, -np.inf]) # both +/- infinity present
nan