创建一个Python字典，其中包含来自另一个字典的特定键值对

Question

我有一些代码是这样的：

d = {'foo': True, 'bar': 42, 'baz': '!'}

a = {'foo': d['foo'], 'bar': d['bar']}
b = {'foo': d['foo'], 'baz': d['baz']}
c = {'bar': d['bar'], 'baz': d['baz']}

当然有更好的方式来表达这一点。实际上，我阅读这些文档是希望字典的copy方法接受要包含在新字典中的键：

# I'd hoped that something like this would work...
a = d.copy('foo', 'bar')
b = d.copy('foo', 'baz')
c = d.copy('bar', 'baz')

我可以为此编写一个函数：

copydict = lambda dct, *keys: {key: dct[key] for key in keys}

a = copydict(d, 'foo', 'bar')
b = copydict(d, 'foo', 'baz')
c = copydict(d, 'bar', 'baz')

有比上面更好的解决方案吗？

Answer 1

我唯一要做的改进是使用真正的函数定义，而不是lambda：

def copy_dict(d, *keys):
    """Make a copy of only the `keys` from dictionary `d`."""
    return {key: d[key] for key in keys}

处理丢失的键可能很有用，但在Python 2中，不能很好地将可选的关键字参数与*参数混合，因此可能需要转到元组参数：

def copy_dict(d, keys, default=None):
    """Make a copy of only the `keys` from dictionary `d`.

    Use `default` if a key is missing.

    """
    return {key: d.get(key, default) for key in keys}

Answer 2

我想问题中的函数是最好的解决方案。

当然，有人可以发布一些惯用的代码，但我相信它不会工作得更好/更快。遍历一个列表并按键从dict中获取项是尽可能快的。

一个建议是从keys参数中删除星号。参数解包增加了不必要的开销。将这些键作为元组传递应该没有问题。

Answer 3

正如@BlaXpirit所指出的，这个解决方案在速度和可读性方面可能不会比您的更好，但是您可以这样做：

>>> from operator import itemgetter
>>> d = {'foo': True, 'bar': 42, 'baz': '!'}
>>> keys = ['foo', 'bar']
>>> dict(zip(keys, itemgetter(*keys)(d)))
{'bar': 42, 'foo': True}