Создать словарь 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

Я думаю, что функция в вопросе - лучшее решение.

Конечно, кто-то может опубликовать какой-то идиоматический код, но я уверен, что он не будет работать лучше / быстрее. Итерирование по списку и получение элементов из запроса по ключу выполняется настолько быстро, насколько это возможно.

Одно из предложений - убрать звезду из параметра keys. Распаковка аргумента добавляет ненужные накладные расходы. Не должно быть проблем с передачей этих ключей в виде кортежа.

Answer 2

Единственное улучшение, которое я хотел бы сделать, это использовать определение реальной функции, а не лямбду:

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

Как отмечает @BlaXpirit, это решение, вероятно, не будет лучше вашего с точки зрения скорости или читабельности, но вы можете сделать что-то вроде этого:

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