python - 在 Python 中有效地按键选择字典项目

Question

假设我有一个字典，它的键是字符串。我如何才能有效地从仅包含某些列表中存在的键的字典中创建新字典？

例如:

# a dictionary mapping strings to stuff
mydict = {'quux': ...,
          'bar': ...,
          'foo': ...}

# list of keys to be selected from mydict
keys_to_select = ['foo', 'bar', ...]

我想出的方法是:

filtered_mydict = [mydict[k] for k in mydict.keys() \ 
                   if k in keys_to_select]

但我认为这是非常低效的，因为:(1) 它需要使用 keys() 枚举键，(2) 它需要每次在 keys_to_select 中查找 k。我认为至少可以避免其中一个。有任何想法吗？如果需要，我也可以使用 scipy/numpy。

Answer 1

dict((k, mydict[k]) for k in keys_to_select)

如果你知道所有要选择的键也是mydict中的键；如果不是这样，

dict((k, mydict[k]) for k in keys_to_select if k in mydict)