使用装饰器与迭代来设置值？

Question

所以我必须遍历一个对象列表，使用它们的一些值进行计算，然后为它们分配新值。

因为列表中的许多项目都将被分配相同的新值，所以我使用字典来保存需要相同值的项目列表。例如：

item_dict = {}

for item in list:
    value = item.value
    if value not in item_dict:
        item_dict[value] = [item]
    else:
        item_dict[value].append(item)

# do some calculations base on values

new_data # some dictionary created by computation
# new data is stored new_data[value] = new_value

for value, new_value in new_data.items():
    items = item_dict[value]
    for item in items:
        item.value = new_value

我正在考虑使用装饰器删除 items 循环中的 for 项目，因为该列表的所有 new_value(s) 都是相同的。例如：

def dec(item):
    def wrap(value):
        item.value = value
    return wrap

def rec(item, func):
    def wrap(value):
        item.value = value
        func(value)
    return wrap

item_dict = {}

for item in list:
    value = item.value
    if value not in item_dict:
        item_dict[value] = dec(item)
    else:
        item_dict[value] = rec(item, item_dict[value])

# do some calculations base on values

new_data # some dictionary created by computation
# new data is stored new_data[value] = new_value

for value, new_value in new_data.items():
    items = item_dict[value]
    items(new_value)

装饰时尚会更有效率吗？它会对记忆产生多大的影响？有没有更好的方法来做到这一点？

Answer 1

一个 defaultdict 在这里工作得很好：

from collections import defaultdict

item_dict = defaultdict(list)

for item in value_list:
    item_dict[item.value].append(item)

# do some calculations base on values

new_data # some dictionary created by computation
# new data is stored new_data[value] = new_value

for value, new_value in new_data.items():
    for item in item_dict[value]:
        item.value = new_value

我很难想出装饰器版本可以更好的方法 - 一方面，您必须担心递归限制。

Answer 2

get 方法在第一种情况下效果很好。

item_dict = {}

for item in list:
    item_dict[item.value] = item_dict.get(item.value, []) + [item]

完成这项工作的关键是使用列表添加而不是追加，因为追加返回 None。