Python numpy.fromfunction()用法及代碼示例

numpy.fromfunction()函數通過在每個坐標上執行一個函數來構造一個數組，因此，所得數組在坐標(x，y，z)處的值為fn(x，y，z)。

Syntax:numpy.fromfunction(function, shape,  dtype)

Parameters:

function:[callable] The function is called with N parameters, where N is the rank of shape. Each parameter represents the coordinates of the array varying along a specific axis.

shape:[(N, ) tuple of ints] Shape of the output array, which also determines the shape of the coordinate arrays passed to function.

dtype:[data-type, optional] Data-type of the coordinate arrays passed to function.

Return:The output array.

代碼1：

# Python program explaining 
# numpy.fromfunction() function 
    
# importing numpy as geek 
import numpy as geek 
    
gfg = geek.fromfunction(lambda i, j:i * j, (4, 4), dtype = float) 
    
print(gfg)

輸出：

[[0. 0. 0. 0.]

[0. 1. 2. 3.]

[0. 2. 4. 6.]

[0. 3. 6. 9.]]

代碼2：

# Python program explaining 
# numpy.fromfunction() function 
    
# importing numpy as geek 
import numpy as geek 
    
gfg = geek.fromfunction(lambda i, j:i == j, (3, 3), dtype = int) 
    
print(gfg)

輸出：

[[ True False False]

[False True False]

[False False True]]