當前位置: 首頁>>代碼示例 >>用法及示例精選 >>正文

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。非經特殊聲明，原始代碼版權歸原作者所有，本譯文未經允許或授權，請勿轉載或複製。