堆栈xarray DataArray

Question

我有N个1D xr.DataArray ，其中1个array坐标b和1个scalar坐标a 。 我想将它们合并到2D DataArray有array坐标b ， a 。 这个怎么做？ 我努力了：

x1 = xr.DataArray(np.arange(0,3)[...,np.newaxis], coords=[('b', np.arange(3,6)),('a', [10])]).squeeze()
x2 = xr.DataArray(np.arange(0,3)[...,np.newaxis], coords=[('b', np.arange(3,6)),('a', [11])]).squeeze()

xcombined = xr.concat([x1, x2])
xcombined

结果是 ：

<xarray.DataArray (concat_dims: 2, b: 3)>
array([[0, 1, 2],
       [0, 1, 2]])
Coordinates:
  * b        (b) int64 3 4 5
    a        (concat_dims) int64 10 11
Dimensions without coordinates: concat_dims

现在，我想选择一个特别的“ a”：

xcombined.sel(a=10)

但是，这引起了：

ValueError: dimensions or multi-index levels ['a'] do not exist

Answer 1

如果您为concat提供dim ，则可以使用：

xcombined = xr.concat([x1, x2], dim='a')

接着：

xcombined.sel(a=10)

<xarray.DataArray (b: 3)>
array([0, 1, 2])
Coordinates:
  * b        (b) int64 3 4 5
    a        int64 10