在Python中,可以使用NumPy库来操作多维数组。如果想用一个函数的返回值填充一个NumPy数组,可以使用NumPy的fromfunction()函数。
fromfunction()函数的语法如下: np.fromfunction(function, shape, **kwargs)
其中,function是一个函数,接受数组的坐标作为输入,并返回相应位置的值;shape是一个元组,用来指定生成的数组的形状;kwargs是用来传递给function的额外参数。
以下是一个示例代码,演示如何使用函数值填充np数组:
import numpy as np
def my_function(x, y):
return x + y
shape = (3, 3) # 定义数组形状为3行3列
arr = np.fromfunction(my_function, shape)
print(arr)
运行结果如下:
[[0. 1. 2.]
[1. 2. 3.]
[2. 3. 4.]]
在这个例子中,my_function函数接受两个参数x和y,并返回它们的和。fromfunction()函数根据指定的形状创建了一个数组,并将my_function的返回值填充到数组中。
希望这个答案能满足你的要求。如果你对其他云计算相关的问题有兴趣,可以继续提问。
领取专属 10元无门槛券
手把手带您无忧上云