Python生成器迭代问题：如何迭代至满足指定条件时停止

问题分析与原理拆解 #

先聊聊你第一次写的代码为啥会出问题：

gen = test_gen()
for i in gen:
    while i > 10:
        print(i)

这里的核心矛盾是**while循环的逻辑和生成器的迭代机制完全不匹配**：

• for i in gen会自动帮你迭代生成器：每次调用生成器的__next__()方法，执行到下一个yield拿到返回值，赋值给i，然后暂停生成器，进入循环体。
• 当i是1、2、5、8时，i > 10为假，while循环直接跳过，for循环继续取下一个值。
• 等i取到12时，i > 10为真，进入while循环，但这个循环里没有任何代码会改变i的值，也不会让生成器继续推进（你的生成器已经走到最后一个yield了），所以会一直打印12，陷入死循环（你说的“无限打印1”应该是笔误啦😉）。

而你第二次用if的代码就正常了：

gen = test_gen()
for i in gen:
    if i > 10:
        print(i)

这是因为if是一次性判断：拿到当前i后，判断是否大于10，是就打印，然后for循环继续取下一个值——但你的生成器在yield 12之后就没有更多值了，for循环会自动捕获StopIteration异常并终止，自然不会出现死循环。

生成器迭代的核心逻辑 #

要搞懂这个问题，得先摸透Python生成器的迭代本质：

• 调用生成器函数（比如test_gen()）不会立即执行函数里的代码，只会返回一个生成器对象。
• 只有对生成器调用__next__()方法（for循环会自动做这件事），生成器才会执行到下一个yield语句，返回对应的值，然后暂停在这个位置。
• 当生成器执行完所有代码（没有更多yield），会抛出StopIteration异常，for循环捕获这个异常后就会自动停止。

符合需求的规范实现 #

根据你的描述“持续迭代该生成器并打印值，直到遇到大于10的数值为止”，我猜你是想打印所有生成的数值，直到碰到大于10的那个，然后停止（包括打印这个大于10的值）？如果是这样，更贴合需求的写法应该用break终止循环：

gen = test_gen()
for i in gen:
    print(i)
    if i > 10:
        break

这样会依次打印1、2、5、8、12，然后因为i>10触发break，循环直接终止，完美匹配“直到遇到大于10的数值为止”的需求。

如果你的需求只是打印大于10的数值（忽略前面小于等于10的），那你第二次的代码已经是正确的，还可以用更简洁的写法，比如结合filter：

gen = test_gen()
for i in filter(lambda x: x > 10, gen):
    print(i)

这样会直接过滤出大于10的值并打印，代码更简洁优雅。

内容的提问来源于stack exchange，提问作者TheStrangeQuark