python - 获取 Python 字典的子集

Question

我有一本字典:

{'key1':1, 'key2':2, 'key3':3}

我需要将该字典的子集传递给第三方代码。它只想要一个包含键 ['key1', 'key2', 'key99'] 的字典，如果它得到另一个键(例如 'key3')，它会爆炸一团糟。有问题的代码超出了我的控制范围，所以我不得不清理我的字典。

将字典限制为一组键的最佳方法是什么？

鉴于上面的示例字典和允许的键，我想要:

{'key1':1, 'key2':2}

Answer 1

In [38]: adict={'key1':1, 'key2':2, 'key3':3}
In [41]: dict((k,adict[k]) for k in ('key1','key2','key99') if k in adict)
Out[41]: {'key1': 1, 'key2': 2}

在 Python3(或 Python2.7 或更高版本)中，您可以使用 dict-comprehension太:

>>> {k:adict[k] for k in ('key1','key2','key99') if k in adict}
{'key2': 2, 'key1': 1}