为什么Python带yield的生成器内的while循环没有按预期终止？

两个核心认知偏差导致实际行为和预期脱节 #

• 对while循环的判断时机理解错误：循环条件仅在每一轮循环体启动执行前做一次求值，循环体内部修改变量值时，不会立刻中断当前正在执行的循环代码，必须等当前循环块的所有代码按顺序执行完毕，回到循环头部时，才会基于最新的变量值判断是否要终止循环。
• 对代码执行顺序的漏算：在把n赋值为0的语句之后，紧跟了一行n +=1。也就是说当你执行完n=0后，代码不会立刻跳回循环判断逻辑，而是会顺序执行完当前循环块剩下的所有代码，直接把刚赋值的0重新修改为1。等真正回到while n:的判断位置时，拿到的n值是1（布尔值为真），自然不会触发循环终止。

逐流程模拟（以初始传入参数n=1为例） #

1. 循环头部判断：n=1为真值，进入循环体
   • 判断n>10：1不满足条件，跳过if块
   • n自增为2，执行yield 2
2. 下一次调用生成器时，回到循环头部判断：n=2为真值，进入循环体
   • 判断n>10：2不满足条件
   • n自增为3，执行yield 3
     ...（中间n从3持续自增到11的过程省略）
     N. 当循环头部进入时的n值为11：
   • 判断n>10：11满足条件，进入if块执行n = 0
   • 顺序执行下一行代码：n +=1 → n从0变为1
   • 执行yield 1
   • 回到循环头部判断条件：n=1为真值，继续执行下一轮循环
     → 至此进入无限循环周期：n会从1一直递增到11，触发条件后被置0再自增为1，重复上述流程永远不会停止。

常见修正方向 #

如果预期是“n超过10时置0就终止循环”，只需要调整代码逻辑，避免n被置0后又被修改为非0值即可：

• 方案1：把自增操作移到if判断之前，保证n被置0后不会被后续代码篡改

def gen(n):
    while n:
        n +=1
        if n >10:
            n =0
        yield n

• 方案2：当满足n>10的条件时直接终止循环，不再执行后续自增和yield逻辑

def gen(n):
    while n:
        if n >10:
            break
        n +=1
        yield n

注意：只要n被置0之后，还有任何修改n值的代码，“n为0就终止循环”的预期就会落空——因为等代码走到循环头部做条件判断时，n早就不是0了。

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