使用 send 将值传递给 yield

Question

当我在谷歌上搜索 python yield 时，我发现了一些有趣的东西，我以前从未知道过，即我们可以将值传递给 yield 以更改 next() 值。我希望一些新的 pythonist 现在可能会意识到这一点，我也不确定它是如何工作的。所以，如果有任何机构可以解释它是如何工作的，以及当我使用send() 发送新索引以产生时行为如何变化。

这是我的 sn-ps：

def print_me (count):
    i = 0
    while i < count:
        val = (yield i)
        if val is not None:
            i = val
        else:
            i += 1

gen = print_me(10)

for i in gen:
   if i == 5:
       gen.send(8)
   else:
       print i

这是我的输出：

0
1
2
3
4
9

Answer 1

您的测试代码实际上很有启发性。在像这样的生成器函数中发生的情况是，每当您在生成器上调用 next 时，代码都会被执行，直到遇到 yield 语句。产生该值，然后在生成器中“暂停”执行。如果您向生成器.send 一些数据，则该值将从yield返回。如果你没有 .send 任何东西，yield 将返回 None。

所以在你的代码中，当你 .send(8) 时，它会在你的生成器中设置 val = 8 (val = yield i)。因为val is not None，i 设置为8。 .send 实际上一直执行到下一个 yield 语句（返回该值 -- 8）。然后一切恢复正常（下一个要生成的数字是9）。