符合子发电机的发电机

Question

我喜欢让发电机顺从其他发电机,例如

def gx():
    for i in [1, 2, 3]:
        yield i

def gy():
    for i in [11, 12, 13]:
        yield i

def gz():
    """this should defer to gx and gy to
       generate [1, 2, 3, 11, 12, 13]"""
    for i in gx(): yield i
    for i in gy(): yield i

gz()中的显式循环是实现此目的的唯一方法,还是有更好的替代方案？

Answer 1

在当前发布的Python版本中,显式循环是调用子生成器的唯一方法.(我认为你的例子只是一个例子 - 不是你要解决的确切问题.)

Python 3.3将yield from为此目的添加特殊语法:

def gz():
    """this should defer to gx and gy to
       generate [1, 2, 3, 11, 12, 13]"""
    yield from gx()
    yield from gy()

有关详细信息,请参阅PEP 380.

Answer 2

使用itertools.chain:

import itertools

gz = itertools.chain(gx(), gy())

在chain他们的文档中,他们通过实现来描述它:

def chain(*iterables):
    for it in iterables:
        for element in it:
            yield element

你也可以从中汲取灵感:)