python - 打包/解包发电机

Question

我有两个发电机。第一个生成器有时需要调用第二个生成器并返回它从那里获得的值:

def a():
    for _b in b():
        yield _b

def b():
    yield 1
    yield 2

for _a in a():
    print _a

有没有更优雅的方式来做到这一点:

for _b in b():
    yield _b

我已经试过了:

yield *b()

但肯定是行不通的。我有 Python 2.6。

Answer 1

我想你的意思是PEP380 .这是available from Python 3.3 .看起来像:

 yield from b()

在 Python2 中没有特殊的语法。您只需使用一个 for 循环。

您问题中的 a 函数实际上完全没有用。您可以在原处使用 b。