Python numpy.fromfunction()函数

numpy.fromfunction()函数通过在每个坐标上执行一个函数来构造一个数组，因此，产生的数组在坐标(x, y, z)上有一个值fn(x, y, z)。

语法: numpy.fromfunction(function, shape, dtype)

参数 :

function : [可调用] 该函数被调用时有N个参数，其中N是形状的等级。每个参数代表沿特定轴线变化的数组的坐标。

shape : [(N, ) tuple of ints] 输出阵列的形状，它也决定了传递给函数的坐标阵列的形状。

dtype : [data-type, optional] 传给函数的坐标阵列的数据类型。

返回：输出数组。

代码 #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)

输出 :