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/')