numpy.fromfunction

numpy.fromfunction(function, shape, **kwargs)

通过在每个坐标上执行函数来构造数组。

The resulting array therefore has a value fn(x, y, z) at coordinate (x, y, z).

参数：

函数：callable

该函数用N个参数调用，其中N是shape的秩。每个参数表示沿着特定轴变化的数组的坐标。例如，如果shape为（2， 2），则参数依次为（0,0 ），（0,1），（1,0），（1,1）。

shape：（N，）ints的元组

输出数组的形状，也决定了传递给函数的坐标数组的形状。

dtype：数据类型，可选

传递给函数的坐标数组的数据类型。默认情况下，dtype是float。

返回：

fromfunction：any

调用函数的结果直接返回。因此，fromfunction的形状完全由函数确定。如果函数返回标量值，则fromfunction的形状将匹配shape参数。

也可以看看

indices，meshgrid

笔记

除dtype之外的关键字传递到函数。

例子

>>> np.fromfunction(lambda i, j: i == j, (3, 3), dtype=int)
array([[ True, False, False],
       [False,  True, False],
       [False, False,  True]], dtype=bool)

>>> np.fromfunction(lambda i, j: i + j, (3, 3), dtype=int)
array([[0, 1, 2],
       [1, 2, 3],
       [2, 3, 4]])