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


Python dask.array.stack用法及代碼示例


用法:

dask.array.stack(seq, axis=0, allow_unknown_chunksizes=False)

沿新軸堆疊數組

給定一係列 dask 數組,通過沿新維度堆疊它們來形成一個新的 dask 數組(默認情況下,axis=0)

參數

seq: list of dask.arrays
axis: int

對齊所有數組的維度

allow_unknown_chunksizes: bool

允許未知的塊大小,例如來自從 dask 數據幀轉換而來的。 Dask.array 無法驗證塊是否對齊。如果數據來自不同對齊的來源,那麽這可能會導致意外結果。

例子

創建切片

>>> import dask.array as da
>>> import numpy as np
>>> data = [da.from_array(np.ones((4, 4)), chunks=(2, 2))
...         for i in range(3)]
>>> x = da.stack(data, axis=0)
>>> x.shape
(3, 4, 4)
>>> da.stack(data, axis=1).shape
(4, 3, 4)
>>> da.stack(data, axis=-1).shape
(4, 4, 3)

結果是一個新的 dask Array

相關用法


注:本文由純淨天空篩選整理自dask.org大神的英文原創作品 dask.array.stack。非經特殊聲明,原始代碼版權歸原作者所有,本譯文未經允許或授權,請勿轉載或複製。