Python生成式的send()方法详解

什么是Python生成器

简单来说，Python中的生成器(generator)是一种特殊的函数，它使用yield关键字来代替return关键字返回自己的值，当函数被调用时，在yield语句处暂停，等待下一次使用。生成器每次返回的值都是可迭代的.

生成器具有如下特点：

1. 生成器是迭代器。
2. 不需要像普通函数一样使用return语句返回一个结果，在生成器中，我们可以使用yield语句来返回。
3. 每次迭代时，生成器会从上一次yield语句处继续执行。

Python生成式中的send()方法

在Python的生成器中，除了yield语句之外，还有一个非常重要的关键字send()。send()方法用于向生成器发送值，同时也是用来启动生成器的。

语法

send()方法的语法如下：

generator.send(value)

其中value表示要发送到生成器里的值。

说明

1. send()方法启动生成器，同时还可以向生成器里发送值。
2. send()方法需要将生成器前一个yield语句接受的值当做返回值返回，同时也是为了替换生成器中yield表达式的值。
3. 如果在生成器的初始位置调用send方法，那么只能发送None，否则会抛出异常。

示例1

def countdown(n):
    while n > 0:
        yield n
        n -= 1

c = countdown(5)   
c.send(None)  # None
c.send("start counting down!")
for i in c:
    print(i)  # 输出 5, 4, 3, 2, 1

在上面的代码中，我们定义了一个计数器生成器，从n开始，每次迭代减1直到0。在第一次调用send()方法之前，生成器必须被调用一次，来启动生成器。在第一次调用send()方法时，必须发送None，之后每次使用send方法来发送一个新的值，这个值将会作为上次yield语句所返回的值。在上面的代码中，第二次使用send()方法发送start counting down!并将其作为返回值返回。此后，迭代器会继续生成从5到1的数字。

示例2

def generator():
    while True:
        n = yield
        print(n)

g = generator()
next(g)

g.send("Hello, World!")
g.send("Python generator tutorial")

在上面的代码中，我们定义了一个生成器，此生成器的作用只是接受一个值并将其打印出来。在第一次调用next()方法后，生成器暂停在yield语句处，等待被唤醒。我们传递的第一个值将被忽略，然后第二个值"Hello, World!"被接受并打印到控制台。在第三个send()调用中，我们又向生成器发送了另一个文本，会将其打印到控制台。

结论

本文中我们提到了Python中的生成器和生成器的send()方法。生成器是一种特别的函数， 使用yield语句替代return语句返回值。send()方法用于向生成器里发送值，启动生成器并将上一个yield语句的结果替换成send的参数，每次迭代时从上一个yield语句处继续执行。了解了这些内容后，你可以在编写Python程序中灵活使用生成器和send()方法，以提升你的Python编程技能。