Могу ли я запомнить генератор Python? - PullRequest
Новая
       15

Могу ли я запомнить генератор Python?

22 голосов
/ 31 декабря 2010

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

Как я могу изменить декоратор памятки, который работает для генераторов Python?Я понимаю, что в какой-то момент мне нужно будет сохранить его в памяти, но я бы хотел обработать это в декораторе, а не изменять исходную функцию.

Текущий код функции запоминания: 

def memoized(f):
    # Warning: Doesn't work if f yields values
    cache={}
    def ret(*args):
        if args in cache:
            return cache[args]
        else:
            answer=f(*args)
            cache[args]=answer
            return answer
    return ret

Ответы [ 4 ]

20 голосов
/ 23 мая 2012

Я понимаю, что это несколько старый вопрос, но для тех, кто хочет получить полное решение: вот один, основанный на предложении jsbueno: 

from itertools import tee
from types import GeneratorType

Tee = tee([], 1)[0].__class__

def memoized(f):
    cache={}
    def ret(*args):
        if args not in cache:
            cache[args]=f(*args)
        if isinstance(cache[args], (GeneratorType, Tee)):
            # the original can't be used any more,
            # so we need to change the cache as well
            cache[args], r = tee(cache[args])
            return r
        return cache[args]
    return ret
10 голосов
/ 31 декабря 2010 
from itertools import tee

sequence, memoized_sequence = tee (sequence, 2)

Готово.

Это проще для генераторов, потому что в стандартной библиотеке есть этот метод "tee"!

4 голосов
/ 31 декабря 2010

Да. Здесь размещен декоратор здесь . Обратите внимание, что, как написано на плакате, вы теряете часть пользы от ленивой оценки. 

def memoize(func):
    def inner(arg):
        if isinstance(arg, list):
            # Make arg immutable
            arg = tuple(arg)
        if arg in inner.cache:
            print "Using cache for %s" % repr(arg)
            for i in inner.cache[arg]:
                yield i
        else:
            print "Building new for %s" % repr(arg)
            temp = []
            for i in func(arg):
                temp.append(i)
                yield i
            inner.cache[arg] = temp
    inner.cache = {}
    return inner


@memoize
def gen(x):
    if not x:
        yield 0
        return

    for i in xrange(len(x)):
        for a in gen(x[i + 1:]):
            yield a + x[0]


print "Round 1"
for a in gen([2, 3, 4, 5]):
    print a

print
print "Round 2"
for a in gen([2, 3, 4, 5]):
    print a
0 голосов
/ 22 ноября 2018

Аналогично другим ответам, но проще, если вы знаете , что f является генератором: 

def memoized_generator(f):
    cache = {}
    @functools.wraps(f)
    def wrapper(*args, **kwargs):
        k = args, frozenset(kwargs.items())
        it = cache[k] if k in cache else f(*args, **kwargs)
        cache[k], result = itertools.tee(it)
        return result
    return wrapper
...