numpy.fromfunction

原文：https://docs.scipy.org/doc/numpy/reference/generated/numpy.fromfunction.html

译者：飞龙 UsyiyiCN

校对：（虚位以待）

numpy.fromfunction(function, shape, **kwargs)[source]

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

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]])