Python numpy fromfunction用法及代碼示例

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

用法:

numpy.fromfunction(function, shape, *, dtype=<class 'float'>, like=None, **kwargs)

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

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

參數：

function： 可調用的

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

shape： (N,) 整數元組

輸出數組的形狀，它也決定了傳遞給函數的坐標數組的形狀。

dtype： 數據類型，可選

傳遞給的坐標數組的數據類型函數.默認，numpy.dtype是浮點數的。

like： array_like

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

返回：

fromfunction： 任何

調用結果函數直接傳回去。因此形狀為fromfunction完全由函數.如果函數返回一個標量值，形狀fromfunction不匹配numpy.shape範圍。

注意：

關鍵字以外的numpy.dtype被傳遞給函數.

例子：

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