使用dask.delayed坐标创建xr.DataArray遇维度冲突问题求助

直接把延迟的坐标包装成dask数组，就能让xarray保留它的延迟状态，不会提前触发计算：

1. 用dask.array.from_delayed()转换延迟坐标

   import dask.array as da
   # 将delayed_coords转为dask数组，shape用None表示维度未知，dtype可提前指定或让dask自动推断
   dask_coords = da.from_delayed(delayed_coords, shape=(None,), dtype=delayed_coords.dtype)
   

2. 用转换后的dask数组创建DataArray

   data_array = xr.DataArray(delayed_values, dims=['pulseId'], coords={'pulseId': dask_coords})
   

关键原理 #

xarray遇到dask数组时会保留其延迟计算特性，不会立即解析维度信息。只有当你主动调用.compute()或者需要实际数据输出时，才会同时触发数据和坐标的计算，这时候二者的维度会同步确定，就不会出现维度不匹配的报错了。

内容的提问来源于stack exchange，提问作者pnjun