python - 用另一个坐标替换xarray坐标

我尝试通过对两个坐标使用一个坐标来替换xarray.Dataset中具有相同长度的两个坐标。我不在乎旧的坐标或它的值。我只是想替换它们。有什么方便的方法吗？

import xarray as xr
ds = xr.Dataset({
    "a": ("a_coord", [0, 1, 2]),
    "b": ("b_coord", [50, 10, 20]),
    "a_coord": [4, 8, 9],
    "b_coord": [3, 1, 4],
    "new_coord": [0, 1, 2],
})
# Get rid of a_coord and b_coord and replace both with new_coord, so a and b
# have new_coord as coordinate.
# But how?

我尝试了rename，但是它返回冲突错误（因为new_coord已经存在）。

ds.rename({
    "a_coord": "new_coord",
    "b_coord": "new_coord",
})

我是否必须将新坐标显式分配给每个依赖于旧坐标之一的数据变量？由于我有许多数据变量（不仅是a和b）以及其他坐标，因此可能会变得混乱。

最佳答案

您需要先重置索引：

ds.reset_index(['a_coord', 'b_coord'], drop = True, inplace = True)
ds['a']= ds.a.rename({'a_coord': 'new_coord'})
ds['b']= ds.b.rename({'b_coord': 'new_coord'})

关于python - 用另一个坐标替换xarray坐标，我们在Stack Overflow上找到一个类似的问题：https://stackoverflow.com/questions/49853326/