xarray DataArray中尺寸的一组坐标

Question

我喜欢Concat的/堆栈不同DataArrays成一个新的，与在所述的concat尺寸一定坐标：

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

xc = xr.concat([x1,x2], dim='new')

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

如何设置维度的坐标与concat函数“新”？

Answer 1

在dim功能concat参数可以在除CONCAT维度的名称DataArray中或pandas.Index：

xc = xr.concat([x1,x2], dim=xr.DataArray(['a', 'b'], dims='new', name='new'))

在xarray.concat文档的详细信息：http://xarray.pydata.org/en/stable/generated/xarray.concat.html#xarray.concat

Answer 2

设定坐标可以通过这样做：

xc['new'] = ['a', 'b']