多维 xarray 在哪里 dorps 坐标？

如何解决多维 xarray 在哪里 dorps 坐标？

我有一个多维变量，我想使用多维掩码对其进行掩码。但是，使用 xarray.where 会删除变量的坐标。这是一个使用非常小的文件的示例（链接到下载 var 文件 here 和掩码文件 here）：

import xarray as xr
mask = xr.open_dataarray("mask.nc")
var = xr.open_dataarray("var.nc")

masked_var = var.where(mask,drop=False)

>>> print(mask.shape)
(3,3,101,101)
>>> print(var.shape)
(3,101)
>>> print(masked_var.shape)
(1,101)  # Note expected dimension is (3,101)

如您所见，where 已删除坐标。 我的预期行为是：

var.isel(lat=0,lon=0).where(mask.isel(lat=0,lon=0))
var.isel(lat=1,lon=1).where(mask.isel(lat=1,lon=1))
.
.
.
#for all lat and lon,resulting in masked_var with (3,101) shape.