python的generator

def simple_generator():
    # 生成器函数，使用 yield 关键字来返回值
    yield 1
    yield 2
    yield 3

# 创建生成器对象
gen = simple_generator()

# 使用 next() 函数获取生成器的下一个值
print(next(gen))  # 输出: 1
print(next(gen))  # 输出: 2
print(next(gen))  # 输出: 3

# 尝试再次获取下一个值会引发 StopIteration 异常
# print(next(gen))  # 这行代码会引发异常

# 可以使用 for 循环遍历生成器中的所有值
for value in simple_generator():
    print(value)  # 输出: 1 2 3

# 生成器表达式
gen_expr = (x * x for x in range(5))
for value in gen_expr:
    print(value)  # 输出: 0 1 4 9 16

解释说明：

• 生成器函数：通过 yield 关键字定义的函数称为生成器函数。每次调用 next() 时，生成器函数会执行到下一个 yield 语句，并返回该 yield 的值。
• 生成器对象：调用生成器函数会返回一个生成器对象，该对象可以在需要时逐步生成值。
• StopIteration 异常：当生成器没有更多值可返回时，调用 next() 会引发 StopIteration 异常。
• 生成器表达式：类似于列表推导式，但使用圆括号 () 而不是方括号 []，生成器表达式会在需要时惰性地生成值，而不是一次性生成所有值。