python-xarray: xarray.Dataset.keep?

Question

我有一个相当大的xr.Dataset，其中包含约20个数据变量。我只对其中的两个感兴趣。我看到xr.Dataset.drop可以删除数据集的变量。

我正在寻找保留变量的语法。我尝试了f ['hs'，'t01']，但出现了以下错误：

Traceback (most recent call last): File "/nethome/rxb826/local/bin/miniconda3/lib/python3.6/site-packages/xarray/core/dataset.py", line 662, in _construct_dataarray variable = self._variables[name] KeyError: ('hs', 't01')

如果没有简单的语法，我可以考虑开发xr.Dataset.keep。我也可以手动创建数据集，如下所示：ds = xr.Dataset({'hs': f['hs'], 't01': f['t01']})。

- Ray Bell

Answer 1

您可以使用变量名的可迭代对象来使用getitem语法：

f_with_two_vars = f[['hs','t01']]

（请参见xarray手册中索引和选择数据一节，了解更详细的解释。）

- jhamman