python的生成器对象,在Python中重置生成器对象

I have a generator object returned by multiple yield. Preparation to call this generator is rather time-consuming operation. That is why I want to reuse the generator several times.

y = FunctionWithYield()

for x in y: print(x)

#here must be something to reset 'y'

for x in y: print(x)

Of course, I'm taking in mind copying content into simple list. Is there a way to reset my generator?

解决方案

Another option is to use the itertools.tee() function to create a second version of your generator:

y = FunctionWithYield()

y, y_backup = tee(y)

for x in y:

print(x)

for x in y_backup:

print(x)

This could be beneficial from memory usage point of view if the original iteration might not process all the items.