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

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。非经特殊声明，原始代码版权归原作者所有，本译文未经允许或授权，请勿转载或复制。