Python：生成器-CSDN博客

本文链接：https://blog.csdn.net/qq_39297053/article/details/137610474

注意：本文引用自专业人工智能社区Venus AI

什么是生成器？

生成器是一个返回迭代器的函数，它记住了上次执行时的位置。与普通函数不同，生成器在每次产生一个值后，会挂起其状态。当生成器再次被调用时，它会从上次离开的位置继续执行。

创建生成器

生成器的创建通常是通过在函数中使用yield关键字。当函数包含yield时，它被识别为生成器函数。例如：

def simple_generator():
    yield 1
    yield 2
    yield 3

这个simple_generator函数是一个生成器，每次调用yield时，它会返回一个值并暂停执行。

如何使用生成器？

使用生成器的基本方法是通过for循环或next()函数。

使用for循环

for value in simple_generator():
    print(value)

这个循环会依次打印：

使用next()函数：

gen = simple_generator()
print(next(gen))  # 输出 1
print(next(gen))  # 输出 2
print(next(gen))  # 输出 3

每次调用next()时，生成器从上次暂停的地方继续执行，直到遇到下一个yield。

生成器的优势

内存效率：生成器逐个产生值，不需要一次性将所有值加载到内存中。
惰性求值：生成器只在需要时才计算下一个值，这对于大数据集非常有用。
简化代码：生成器使得编写迭代代码更加简单。

生成器表达式

除了生成器函数外，还可以使用生成器表达式来创建生成器。它类似于列表推导式，但使用圆括号。

gen_expr = (x * 2 for x in range(3))
for val in gen_expr:
    print(val)

这将输出：

高级用法：协程

什么是协程？

协程，有时被称为微线程，是一种计算机程序的组件，它具有自己的执行状态（比如局部变量）和控制流程。与传统的子程序（如函数或方法）不同的是，协程可以在某个点暂停其执行，并在需要时从相同的位置继续执行。

Python中的协程

在Python中，协程是通过生成器实现的。协程与生成器的主要区别在于，协程不仅可以生成值（使用yield），还可以消费值（使用send()方法发送值）。

简单协程示例

让我们通过一个简单的例子来理解协程是如何工作的：

def simple_coroutine():
    print("Coroutine has been started")
    x = yield
    print("Coroutine received:", x)
# 创建协程对象
coro = simple_coroutine()
# 启动协程
next(coro)
# 发送值到协程
coro.send(10)

在这个例子中，simple_coroutine是一个协程函数。当我们第一次调用next(coro)时，协程开始执行并打印出”Coroutine has been started”。当它到达yield语句时，它会暂停，等待一个值被发送到协程。然后，我们使用coro.send(10)来发送值10到协程，协程继续执行并打印出接收到的值。