Python生成器多次调用next()触发StopIteration错误的原因及解决方法

咱们先一步步拆解你的问题，先搞清楚错误产生的原因，再讲具体的解决办法～

1. 为什么会触发StopIteration？ #

生成器的核心特性是惰性生成数据：每次调用next()时，它才会执行到下一个yield语句返回值；当所有yield都执行完毕后，生成器就没有更多值可以返回了，这时候再调用next()，Python就会抛出StopIteration异常，明确告诉你“我已经没东西可以输出啦”。

看你第一段代码：

import time
import numpy as np

def sensor_data_generator(n):
    for _ in range(n):
        reading = {
            "timestamp" : time.time(),
            "temperature" : np.random.uniform(20.0, 25.0),
            "humidity" : np.random.uniform(60.0, 100.0),
            "light" : np.random.uniform(0.0, 100.0)
        }
        yield reading
        time.sleep(1)

sensor_stream = sensor_data_generator(5)

for _ in range(6):
    print(next(sensor_stream))

你创建的生成器只会生成5条数据（因为range(5)控制了循环次数），但你却循环了6次去调用next()——前5次都能正常拿到传感器数据，第6次调用时生成器已经耗尽，自然就触发了StopIteration异常。

而你用for reading in gen:的方式遍历就没问题，是因为Python的for循环内部会自动捕获StopIteration异常，一旦捕获到就会自动终止循环，不会让程序崩溃。

2. 如何处理避免循环崩溃？ #

这里给你几种实用的解决方案：

方法一：优先使用for循环遍历（最推荐） #

就像你已经尝试过的写法，Python帮你封装了异常处理逻辑，代码简洁又安全：

gen = sensor_data_generator(5)

for reading in gen:
    print(reading)

方法二：给next()函数设置默认值 #

next()函数支持传入第二个参数作为默认值，当生成器耗尽时，不会抛出异常，而是返回这个默认值。你可以利用这一点判断是否终止循环：

sensor_stream = sensor_data_generator(5)

for _ in range(6):
    reading = next(sensor_stream, None)
    if reading is None:
        print("生成器已耗尽，停止输出")
        break
    print(reading)

这样第6次调用时会拿到None，你可以选择终止循环或者做其他自定义处理。

方法三：手动捕获StopIteration异常 #

用try-except块包裹next()调用，主动捕获异常并处理，适合需要在异常发生时执行特定逻辑的场景：

sensor_stream = sensor_data_generator(5)

for _ in range(6):
    try:
        reading = next(sensor_stream)
        print(reading)
    except StopIteration:
        print("没有更多传感器数据了")
        break

备注：内容来源于stack exchange，提问作者antn