Python dask.array.nancumsum用法及代碼示例

用法: dask.array.nancumsum(x, axis, dtype=None, out=None, *, method='sequential')

返回給定軸上數組元素的累積總和，將非數字 (NaN) 視為零。當遇到 NaN 並且前導 NaN 被零替換時，累積和不會改變。

此文檔字符串是從 numpy.nancumsum 複製的。

可能存在與 Dask 版本的一些不一致之處。

Dask 添加了一個額外的僅關鍵字參數 method 。

方法{‘sequential’, ‘blelloch’}，可選

選擇用於執行 cumsum 的方法。默認為‘sequential’。

‘sequential’ 在當前塊之前執行每個先前塊的cumsum。
‘blelloch’ 是一個 work-efficient 並行 cumsum。它通過以下方式暴露了並行性
首先取每個塊的總和並通過二叉樹組合總和。根據工作負載、調度程序和硬件，此方法可能更快或更節省內存。更多的基準測試是必要的。

對於 all-NaN 或空的切片返回零。

參數：

a：數組(在 Dask 中不支持): 輸入數組。
axis：整數，可選: 計算累積和的軸。默認值 (None) 是計算展平數組上的 cumsum。
dtype：dtype，可選: 返回數組的類型和對元素求和的累加器的類型。如果未指定 dtype ，則默認為 a 的 dtype，除非 a 具有精度小於默認平台整數的整數 dtype。在這種情況下，將使用默認平台整數。
out：ndarray，可選: 用於放置結果的替代輸出數組。它必須具有與預期輸出相同的形狀和緩衝區長度，但如果需要，類型將被強製轉換。有關更多詳細信息，請參閱輸出類型確定。

nancumsum：數組。: 除非指定了out，否則將返回一個保存結果的新數組，並在其中返回它。如果 axis 不是 None 或 a 是一維數組，則結果的大小與 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.]])

相關用法

注：本文由純淨天空篩選整理自dask.org大神的英文原創作品 dask.array.nancumsum。非經特殊聲明，原始代碼版權歸原作者所有，本譯文未經允許或授權，請勿轉載或複製。

用法:

參數：

返回：

例子：