xarray 中的布尔索引

Question

我有一些带有暗淡 'time', 'lat', 'lon' 的数组和一些只有 'lat', 'lon' 的数组。我经常必须这样做才能使用 2d（经纬度）掩码来屏蔽与时间相关的数据：

x.data[:, mask.data] = np.nan

当然，计算按预期广播。如果y 是 2d lat-lon 数据，它的值会广播到 x 中的所有时间坐标：

z = x + y

但索引并没有像我预期的那样广播。我希望能够做到这一点，但它会引发 ValueError: Buffer has wrong number of dimensions:

x[mask] = np.nan

最后，xr.where 似乎按预期跨时间坐标广播掩码的值，但您不能以这种方式设置值。

x_masked = x.where(mask)

那么，我在这里是否缺少一些有助于使用缺少尺寸（并且需要广播）的布尔掩码设置值的东西？我在顶部提供的选项真的是这样做的方法吗（在这种情况下，我还不如只使用标准的 numpy 数组......）

Answer 1

编辑：这个问题仍在获得支持，但现在要容易得多 - 请参阅this answer

---

这里有点相关的问题：Concise way to filter data in xarray

目前最好的方法是.where 和.fillna 的组合。

valid = date_by_items.notnull()
positive = date_by_items > 0
positive = positive * 2
result = positive.fillna(0.).where(valid)
result

但是 xarray 中的变化将使这更加简洁 - 如果您有兴趣，请查看 GitHub 存储库