dask array.stack
时间: 2023-11-20 08:50:36 浏览: 94
`dask.array.stack`是Dask数组的一个函数,用于将多个Dask数组沿着新的维度堆叠在一起。它的语法如下:
```python
dask.array.stack(arrays, axis=0)
```
其中,`arrays`是一个Dask数组列表,`axis`是新的维度的索引。
例如,我们有两个Dask数组:
```python
import dask.array as da
x = da.ones((2, 3), chunks=(1, 3))
y = da.zeros((2, 3), chunks=(1, 3))
```
我们可以使用`stack`函数将它们沿着新的维度堆叠在一起:
```python
z = da.stack([x, y], axis=0)
```
这将创建一个新的Dask数组`z`,它的形状为`(2, 2, 3)`,其中第一个维度是新的维度,第二个和第三个维度是原来的维度。
阅读全文