Dict comprehension, кортежи и ленивая оценка

Question

Я пытаюсь посмотреть, смогу ли я что-нибудь сфотографировать lazy в Python.

У меня есть определение dict, где значение является кортежем. Я хочу иметь возможность создать вторую запись кортежа, используя первую запись кортежа.

Примером может служить пример.

dictA = {'a': 1, 'b': 3, 'c': 42} 
{key: (a = someComplexFunction(value), moreComplexFunction(a)) for key, value in dictA.items()} 

Возможно ли, что moreComplexFunction использует вычисление в первой записи кортежа?

Answer 1

Вы можете добавить вторую петлю на один-элемент кортежа в:

{key: (a, moreComplexFuntion(a)) for key, value in dictA.items() 
           for a in (someComplexFunction(value),)} 

Это дает вам доступ к выходу someComplexFunction(value) в стоимостном выражении, но это довольно некрасиво.

Лично я бы перейти к регулярной петле в таких случаях:

dictB = {} 
for key, value in dictA.items(): 
    a = someComplexFunction(value) 
    dictB[key] = (a, moreComplexFunction(a)) 

и сделать с ней.

Answer 2

или вы могли бы просто написать функцию, чтобы вернуть кортеж:

def kv_tuple(a): 
    tmp = someComplexFunction(a) 
    return (a, moreComplexFunction(tmp)) 

{key:kv_tuple(value) for key, value in dictA.items()} 

это также дает возможность использовать такие вещи, как namedtuple, чтобы получить имена для элементов кортежа, и т.д. Я не знаю, насколько быстрее/медленнее это будет, хотя ... обычный цикл, скорее всего, будет быстрее (меньше вызовов функций) ...

Answer 3

Наряду с Martijn's answer, используя выражение генератора и понимание dict также является весьма семантическим и ленивым:

dictA = { ... } # Your original dict 

partially_computed = ((key, someComplexFunction(value)) 
         for key, value in dictA.items()) 

dictB = {key: (a, moreComplexFunction(a)) for key, a in partially_computed}