Question

I have some code that looks something like this:

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']}

Surely there’s a better way to express this. I actually read the docs in the hope that a dictionary’s copy method accepts keys to be included in the new dictionary:

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

I could write a function for this purpose:

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')

Is there a better solution than the above?

Answer 1

I guess that the function in the question is the best solution.

Of course, someone could post some idiomatic code, but I’m sure it wouldn’t work better/faster. Iterating over a list and getting items from a dict by key is as fast as it can get.

One suggestion is to remove the star from the keys parameter. Argument unpacking adds unnecessary overhead. There should be no problem with just passing those keys as a tuple.