將值重新分配給字典的pythonic方法

Question

我有一本以列表為值的字典：

my_dict = {1: [2,3], 2: [4, 5], 3: [6, 7]}

我想更新字典以將值更新為舊列表值的總和：

my_dict = {1: 5, 2: 9, 3: 13}

最有效/最有效的方法是什么？ 我通常要做的是：

for key in my_dict:
    my_dict[key] = sum(my_dict[key])

有更好的方法嗎？

Answer 1

您可以使用字典理解：

my_dict = {1: [2,3], 2: [4, 5], 3: [6, 7]}
new_d = {a:sum(b) for a, b in my_dict.items()}

輸出：

{1: 5, 2: 9, 3: 13}

Answer 2

您可以使用reduce代替sum ：

from functools import reduce

my_dict = {1: [2,3], 2: [4, 5], 3: [6, 7]}
final = {k: reduce(lambda x,y: x+y, v) for k,v in my_dict.items()}

輸出：

{1: 5, 2: 9, 3: 13}

否則，您可以參考此線程以獲取更多信息。