Python NumPy fromfunction方法用法及代码示例

Numpy 的 fromfunction(~) 方法使用初始化每个单元格值的函数构造 Numpy 数组。

1. function | callable

一个函数，它接受行和列索引，并返回该单元格的值。

2. shape | sequence 共 int

所得数组的所需形状。

3. dtype | string 或 type | optional

结果数组所需的数据类型。默认情况下，dtype=Float 。

一个 Numpy 数组。

要创建一个 2 x 2 Numpy 数组，其中对角线设置为 True，其他位置设置为 False：

np.fromfunction(lambda i,j: i==j, (2,2))



array([[ True, False],
       [False,  True]])

在这里，该函数接受行索引和列索引作为参数。

在这里，我们定义了一个名为 foo 的显式函数：

def foo(i, j):
 return i + j

我们可以像这样使用fromfunction(~)方法：

np.fromfunction(foo, (3,3))



array([[0., 1., 2.],
       [1., 2., 3.],
       [2., 3., 4.]])

相关用法

注：本文由纯净天空筛选整理自Isshin Inada大神的英文原创作品 NumPy | fromfunction method。非经特殊声明，原始代码版权归原作者所有，本译文未经允许或授权，请勿转载或复制。