从另一个字典创建包含特定键值对的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')

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

相关问题：

• 在Python字典中向键添加多个字典
• 如何在Python 3中将字典追加到字典中？
• 如何获取仅包含一个项的字典中的键值对？
• 在Python中向字典添加值
• 在Python中对字典进行排序
• Python：添加字典项。
• 在Python中重命名字典的键/值
• 在Python中，有类似setdefault()的方法获取对象属性吗？

Answer 1

正如@BlaXpirit所说，这个解决方案在速度或可读性方面可能不如你的方法好，但你可以尝试这样做：

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

Answer 2

我唯一想改进的就是用真正的函数定义，而不是用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中，你不能很好地将可选的关键字参数和*args混合使用，所以你可能需要使用元组作为参数：

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 3

我觉得问题中的这个函数是最好的解决方案。

当然，有人可能会发一些更地道的代码，但我相信那样的代码不会更好或更快。遍历一个列表并通过键从字典中获取项目的速度已经是最快的了。

有一个建议是去掉keys参数前面的星号。参数解包会增加不必要的开销。直接把这些键作为一个元组传进去应该没有问题。