如何在Python中修改生成器？

Question

Python中是否有一个通用接口来修改生成器的行为？

例如,我想修改现有的生成器以在流中插入一些值并删除一些其他值.

我怎么做？

谢谢,Boda Cydo

Answer 1

您可以使用itertools提供的函数来生成生成器并生成新的生成器.

例如,您可以使用takewhile直到不再满足谓词,并链接到一系列新值.

查看其他示例的文档,包括ifilter,dropwhile和islice之类的内容.

Answer 2

您可以将发电机包装在您自己的发电机中.

from itertools import count

def odd_count():
    for i in count():
        if i % 2:
            yield i