python生成器创建的方法整理

Python生成器通过yield、生成器表达式或类实现,可惰性产出值以节省内存。1. 生成器函数用yield返回值并暂停执行;2. 生成器表达式语法简洁,适合简单转换;3. 类实现__iter__和__next__方法,适用于复杂状态管理;4. 支持send、close、throw等高级操作,用于协程与异常控制。

Python生成器是一种特殊的函数,能够按需产生值,节省内存并提升性能。它在遍历大量数据时特别有用。以下是创建Python生成器的几种常用方法整理

1. 使用生成器函数(yield关键字)

最常见的方式是定义一个包含 yield 的函数。调用该函数时返回一个生成器对象,每次迭代时逐个产出值。

  • 函数中使用 yield 而不是 return
  • 每次遇到 yield 暂停执行,保存状态,下次从该位置继续
  • 适合实现惰性计算和无限序列

示例:


def count_up_to(n):
    num = 1
    while num <= n:
        yield num
        num += 1

gen = count_up_to(5) for value in gen: print(value) # 输出 1 到 5

2. 使用生成器表达式

类似于列表推导式,但使用圆括号 (),返回的是生成器对象,不立即占用全部内存。

  • 语法简洁,适合简单数据转换或过滤
  • 不会像列表推导那样一次性生成所有元素

示例:


squares = (x**2 for x in range(1, 6))
for sq in squares:
    print(sq)  # 输出 1, 4, 9, 16, 25

3. 使用类定义生成器(__iter__ 和 __next__)

通过定义类并实现迭代协议,手动控制生成逻辑。适用于需要维护复杂状态的场景。

  • 类中实现 __iter__() 返回自身
  • 实现 __next__() 定义每次返回的值,到末尾抛出 StopIteration
  • 灵活性高,但代码量比函数方式多

示例:


class CountDown:
    def __init__(self, start):
        self.start = start
def __iter__(self):
    return self

def __next__(self):
    if self.start zuojiankuohaophpcn= 0:
        raise StopIteration
    self.start -= 1
    return self.start + 1

gen = CountDown(3) for n in gen: print(n) # 输出 3, 2, 1

4. 生成器中的高级用法

生成器支持一些进阶操作,如传值、关闭、异常处理等。

  • send():向生成器传递值,用于协程场景
  • close():手动关闭生成器,触发 GeneratorExit 异常
  • throw():在生成器内部抛出异常

send() 示例:


def echo_generator():
    while True:
        received = yield
        print(f"Received: {received}")

gen = echo_generator() next(gen) # 启动生成器 gen.send("Hello") # 输出: Received: Hello gen.send("World")

基本上就这些。掌握这几种方法,就能灵活应对各种需要延迟加载或高效处理数据流的场景。不复杂但容易忽略细节,比如必须先调用 next() 才能使用 send()。