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参数。

也可以看看

indicesmeshgrid

笔记

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