当前位置: 首页>>代码示例 >>用法及示例精选 >>正文


Python dask.array.fromfunction用法及代码示例


用法:

dask.array.fromfunction(func, chunks='auto', shape=None, dtype=None, **kwargs)

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

此文档字符串是从 numpy.fromfunction 复制的。

可能存在与 Dask 版本的一些不一致之处。

因此,结果数组在坐标 (x, y, z) 处具有值 fn(x, y, z)

参数

function可调用(在 Dask 中不支持)

该函数使用 N 个参数调用,其中 N 是 shape 的等级。每个参数代表沿特定轴变化的阵列坐标。例如,如果 shape(2, 2) ,那么参数将是 array([[0, 0], [1, 1]])array([[0, 1], [0, 1]])

shape(N,) 整数元组

输出数组的形状,它还确定传递给 function 的坐标数组的形状。

dtype数据类型,可选

传递给 function 的坐标数组的数据类型。默认情况下,dtype 是浮点数。

like数组(在 Dask 中不支持)

引用对象以允许创建不是 NumPy 数组的数组。如果作为like 传入的array-like 支持__array_function__ 协议,则结果将由它定义。在这种情况下,它确保创建一个与通过此参数传入的对象兼容的数组对象。

返回

fromfunction任何

调用function 的结果直接传回。因此 fromfunction 的形状完全由 function 决定。如果 function 返回标量值,则 fromfunction 的形状将与 shape 参数不匹配。

注意

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

相关用法


注:本文由纯净天空筛选整理自dask.org大神的英文原创作品 dask.array.fromfunction。非经特殊声明,原始代码版权归原作者所有,本译文未经允许或授权,请勿转载或复制。