Applying a function to values in dict
If you're using Python 2.7 or 3.x:
d2 = {k: f(v) for k, v in d1.items()}
Which is equivalent to:
d2 = {}
for k, v in d1.items():
d2[k] = f(v)
Otherwise:
d2 = dict((k, f(v)) for k, v in d1.items())
d2 = dict((k, f(v)) for k,v in d.items())
You could use map
:
d2 = dict(d, map(f, d.values()))
If you don't mind using an extension. You can also use valmap
in the toolz library which is functionally equivalent to using the map
solution:
from toolz.dicttoolz import valmap
d2 = valmap(f, d)
If not for the clean presentation of the method, you also have the option of supplying a default return class as well, for people that need something other than a dict
.
Dictionaries can be nested in Python and in this case the solution d2 = {k: f(v) for k, v in d1.items()}
will not work.
For nested dictionaries one needs some function to transverse the whole data structure. For instance if values are allowed to be themselves dictionaries, one can define a function like:
def myfun(d):
for k, v in d.iteritems():
if isinstance(v, dict):
d[k] = myfun(v)
else:
d[k] = f(v)
return d
And then
d2 = myfun(d)