将函数映射到字典中指定键的值

Question

有没有一种方便的方法可以将函数映射到字典中的指定键？

即，给定

d = {"a": 1, "b": 2, "c": 3}

想要将一个函数（比如 f）映射到键“a”和“c”：

{"a": f(1), "b": 2, "c": f(3)}

编辑

寻找不会​​更新输入字典的方法。

Answer 1

您可以使用字典理解：

output_dict = {k: f(v) for k, v in d.items()}

请注意，f(v) 将立即被评估（调用），其返回值将存储为字典的值。

如果你想存储函数并在以后调用它（参数已经存储）你可以使用functools.partial：

from functools import partial


def f(n):
    print(n * 2)


d = {"a": 1, "b": 2, "c": 3}

output_dict = {k: partial(f, v) for k, v in d.items()}

output_dict['b']()
# 4

如果您只想映射特定键，您当然可以不使用.items，而只需覆盖这些键：

d['a'] = partial(f, d['a'])

或更广义的

keys = ('a', 'c')
for key in keys:
    d[key] = partial(f, d[key])