python Xarray DataArray：如何向现有维度添加附加坐标

Question

我刚开始使用Xarray DataArray，我不确定如何将新坐标轴添加到现有维度。谁能指出我正确的方向。

所以我有一个现有的数组，看起来像：

 demo_results = np.zeros([10, 20, 5])
 cols = ['c1', 'c2', 'c3', 'c4', 'c5']
 t = xr.DataArray(demo_results, coords={'sim': cols}, dims=('run', 'year', 'sim'))

所以这会给我一个 3 维数组，沿着“sim”维度有 5 个坐标。现在我想在名为“c6”的“sim”维度上添加一个额外的坐标，但我不知道该怎么做？我在文档中没有看到任何信息或示例。在pandas 中，我只会引用df.loc[:, 'c6] = 0 之类的新维度，这样就可以了。但我用Xarray 尝试过类似的操作，例如：

 t.loc['sim', 'c6'] = 0 

但我出错了：TypeError: invalid indexer array, does not have integer dtype: array('sim', dtype='<U3')

我想我仍然掌握 DataArray 索引的窍门。

Answer 1

Xarray 在底层使用 numpy/dask 数组，因此不支持自动扩展（就像在 Pandas 中一样）。

如果您提前知道列的名称和数量，则可以提前创建整个区域。例如

demo_results = np.zeros([10, 20, 6])
cols = ['c1', 'c2', 'c3', 'c4', 'c5', 'c6']
t = xr.DataArray(demo_results, coords={'sim': cols}, dims=('run', 'year', 'sim'))
t.loc[{'sim': 'c6'}] = -999.  # or whatever value you like

注意，当您使用不支持项目分配的 dask 数组时，这将不起作用。

如果你使用的是 dask，无法提前创建完整的数组，你可以一直使用xarray.concat：

demo_results = np.zeros([10, 20, 1]) - 999.
cols = ['c6']
t6 = xr.DataArray(demo_results, coords={'sim': cols}, dims=('run', 'year', 'sim'))

# concat old+new
t = xr.concat([t, t6], dim='sim')