Возвратите итератор после модификации

Question

У меня есть итератор result в одной функции, и мне нужно вернуть итератор после внесения некоторых изменений в объекты в result из этой функции. Я не хочу перебирать result, так как это приводит к ухудшению производительности. Я хочу выполнить эту операцию «на лету» для каждого элемента в result.

Это код, который выполняет итерацию по результату для выполнения этой операции.

sols = [] 
# Get rid of this loop 
for sol in result: 
    sols.append(dict((symbols[abs(lit) - 1], lit > 0) for lit in sol)) 
# Return iterator object 
return (sol for sol in sols) 

Answer 1

Вы почти у цели. Только не хранить значения в sols списке:

def f(result): 
    return (dict((symbols[abs(lit) - 1], lit > 0) for lit in sol) for sol in result) 

Это дает (не каламбур) генератор, который постепенно проходит данный result параметр.