dask.array.to_npy_stack

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], x[4:5] 的 numpy 数组,这是由沿第零轴的块大小指定的:

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

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

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