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


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


用法:

dask.array.to_npy_stack(dirname, x, axis=0)

将 dask 数组写入一堆 .npy 文件

这会将 dask.array 沿一个轴分区,并将沿该轴的每个块存储为指定目录中的单个 .npy 文件

例子

>>> x = da.ones((5, 10, 10), chunks=(2, 4, 4))  
>>> da.to_npy_stack('data/', x, axis=0)

.npy 文件分别存储 x[0:2], x[2:4], and x[4:5] 的 numpy 数组,由沿第零轴的块大小指定:

$ tree data/
data/
|-- 0.npy
|-- 1.npy
|-- 2.npy
|-- info

info 文件存储数组的 dtype、块和轴信息。您可以使用dask.array.from_npy_stack() 函数加载这些堆栈。

>>> y = da.from_npy_stack('data/')

相关用法


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