Python函数中混用yield与return为何始终返回生成器？

核心原因 #

在Python中，只要函数定义里包含yield关键字，它就会被认定为生成器函数。调用这类函数时，Python不会立即执行函数体代码，而是直接返回一个生成器对象——无论函数内部是否真的走到yield分支。

你的else分支里的return output并不会直接返回列表，而是会在迭代生成器时触发StopIteration异常，output会作为该异常的value属性存在。但由于这个分支没有任何yield语句，迭代生成器时函数体刚执行到return就抛出异常，这就是你遍历时报错的根源。

修复方案 #

要实现“根据列表长度返回生成器或普通列表”的需求，不能在同一个函数里混合生成器逻辑和直接返回列表的逻辑，推荐以下两种方案：

方案1：拆分逻辑，分支处理 #

通过外层函数判断列表长度，分别返回生成器或列表：

def square_result(numbers):
    if len(numbers) > 3:
        # 返回生成器
        def _gen():
            for number in numbers:
                yield number**2
        return _gen()
    else:
        # 返回列表推导式生成的结果
        return [num**2 for num in numbers]

方案2：统一返回可迭代对象（可选） #

如果可以接受长度≤3时也返回生成器（生成器本身也是可迭代对象），可以简化代码：

def square_gen(numbers):
    return (num**2 for num in numbers)

但此方案不符合你原本“返回普通列表”的需求，仅作扩展参考。

验证结果 #

• 调用square_result([1,2,3])会直接返回列表[1,4,9]
• 调用square_result([1,2,3,4])会返回生成器对象，迭代即可得到平方值序列

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