Использование декораторов против итерации для установки значений?

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 

я думаю об удалении для элемента в цикле пунктов с декоратор, так как все new_value (ов) для этого списка являются одинаковыми. Например:

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] 

Ключом к выполнению этой работы является использование добавления списка вместо добавления, поскольку append возвращает None.