當前位置: 首頁>>代碼示例 >>用法及示例精選 >>正文


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


用法:

dask.array.fromfunction(func, chunks='auto', shape=None, dtype=None, **kwargs)

通過在每個坐標上執行一個函數來構造一個數組。

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

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

因此,結果數組在坐標 (x, y, z) 處具有值 fn(x, y, z)

參數

function可調用(在 Dask 中不支持)

該函數使用 N 個參數調用,其中 N 是 shape 的等級。每個參數代表沿特定軸變化的陣列坐標。例如,如果 shape(2, 2) ,那麽參數將是 array([[0, 0], [1, 1]])array([[0, 1], [0, 1]])

shape(N,) 整數元組

輸出數組的形狀,它還確定傳遞給 function 的坐標數組的形狀。

dtype數據類型,可選

傳遞給 function 的坐標數組的數據類型。默認情況下,dtype 是浮點數。

like數組(在 Dask 中不支持)

引用對象以允許創建不是 NumPy 數組的數組。如果作為like 傳入的array-like 支持__array_function__ 協議,則結果將由它定義。在這種情況下,它確保創建一個與通過此參數傳入的對象兼容的數組對象。

返回

fromfunction任何

調用function 的結果直接傳回。因此 fromfunction 的形狀完全由 function 決定。如果 function 返回標量值,則 fromfunction 的形狀將與 shape 參數不匹配。

注意

dtype 以外的關鍵字被傳遞給 function

例子

>>> np.fromfunction(lambda i, j: i == j, (3, 3), dtype=int)  
array([[ True, False, False],
       [False,  True, False],
       [False, False,  True]])
>>> np.fromfunction(lambda i, j: i + j, (3, 3), dtype=int)  
array([[0, 1, 2],
       [1, 2, 3],
       [2, 3, 4]])

相關用法


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