Python生成器如何实现高效迭代

Python生成器是一种特殊的迭代器，它允许你在需要时才生成值，从而节省内存并提高效率

1. 使用yield关键字：在定义生成器函数时，使用yield关键字返回一个值。当生成器函数被调用时，它返回一个生成器对象，而不是直接执行函数体。当生成器对象被迭代时，函数体将被执行，直到遇到yield关键字，此时函数将返回yield后面的值，并暂停执行。下次迭代时，函数将从暂停的地方继续执行，直到再次遇到yield关键字。

def simple_generator():
    yield 1
    yield 2
    yield 3

gen = simple_generator()
for value in gen:
    print(value)

1. 使用for循环迭代：生成器对象可以直接用于for循环进行迭代。

def simple_generator():
    yield 1
    yield 2
    yield 3

for value in simple_generator():
    print(value)

1. 使用next()函数迭代：可以使用next()函数获取生成器对象的下一个值。当生成器对象中没有更多值时，next()函数将引发StopIteration异常。

def simple_generator():
    yield 1
    yield 2
    yield 3

gen = simple_generator()
while True:
    try:
        value = next(gen)
        print(value)
    except StopIteration:
        break

1. 使用send()方法发送值：生成器对象还支持send()方法，允许你在迭代过程中向生成器发送值。当生成器接收到一个值时，它将在下一次yield处返回该值。

def simple_generator():
    x = 0
    while True:
        value = yield x
        if value is not None:
            x = value
        else:
            break

gen = simple_generator()
print(next(gen))  # 输出 0
print(gen.send(5))  # 输出 5
print(next(gen))  # 输出 5

通过这些方法，Python生成器可以实现高效的迭代，节省内存并提高程序性能。