numpy.fromfunction
numpy.fromfunction(function, shape, *, dtype=, **kwargs) [source]
通过在每个坐标上执行一个函数来构造一个数组。
因此,结果数组在坐标(x, y, z)处有一个值fn(x, y, z)。
参数:function :callable
该函数使用N个参数调用,其中N是的秩 shape。
每个参数代表沿特定轴变化的数组坐标。
例如,如果shape 为(2, 2),
则参数将为(2, 2)array([[0, 0], [1, 1]])
和 array([[0, 1], [0, 1]])
shape :(N,)个整数元组
输出数组的形状,它也确定传递给function的坐标数组的形状。
dtype:bool,可选
传递给function的坐标数组的数据类型。默认情况下dtype为float。
返回值:fromfunction :any
调用函数的结果直接传回。因此,的形状fromfunction完全由功能决定 。
如果函数返回标量值,则其形状 fromfunction将与shape参数不匹配。
Notes
关键字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]])