numpy.fromfunction
numpy.fromfunction(function, shape, *, dtype=<class 'float'>, **kwargs) [source]
通过在每个坐标上执行一个函数来构造一个数组。
因此，结果数组在坐标(x, y, z)处有一个值fn(x, y, z)。
参数： | function ：callable 该函数使用N个参数调用，其中N是的秩 shape。 每个参数代表沿特定轴变化的数组坐标。 例如，如果shape 为(2, 2)， 则参数将为 和 shape ： 输出数组的形状，它也确定传递给 dtype： 传递给 |
返回值： | fromfunction ： 调用函数的结果直接传回。因此，的形状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]])