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']})

Answer 1

您可以使用可迭代的变量名来使用 getitem 语法:

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

(有关更详细的解释，请参阅 xarray 手册部分 Indexing and selecting data。)