Python REPL中如何正确调用包含yield关键字的函数？

为什么加了yield后直接调用函数没有反应 #

只要函数内部包含yield关键字，它就不再是普通函数，而是生成器函数：

• 普通函数调用时会直接执行内部代码，遇到return就返回结果并结束运行
• 生成器函数被调用时，不会执行任何内部代码，只会返回一个生成器对象，需要主动迭代这个对象才会触发内部代码执行

你给出的示例代码正确的调用方式如下：

# 第一步：调用生成器函数拿到生成器对象
gen = infinite_sequence()
# 第二步：通过next()迭代生成器，每次迭代会执行到yield就暂停，返回yield后的值
print(next(gen)) # 输出 0
print(next(gen)) # 输出 1
print(next(gen)) # 输出 2

# 也可以用for循环自动迭代，遇到break或者迭代结束就停止
for num in infinite_sequence():
    print(num)
    if num >= 5:
        break

return为什么会影响生成器运行 #

return在生成器函数中的作用和普通函数完全不同：

• 生成器执行过程中遇到return，会直接抛出StopIteration异常，终止整个迭代流程，return后携带的值只会作为异常的参数，不会被普通迭代逻辑直接获取
• 你在生成器里加了return后，迭代会提前终止，所以看起来像是"无法正常运行"

示例验证：

def test_generator():
    yield 1
    return "结束标记"
    yield 3 # 这行代码永远不会执行

g = test_generator()
print(next(g)) # 输出 1
next(g) # 这里会抛出 StopIteration 异常，迭代终止

相关疑问解答 #

• 包含yield/return的函数需要不同的调用方式吗？
  是的，普通函数直接调用即可拿到返回值，生成器函数必须迭代它返回的生成器对象才能触发内部代码执行。
• 这个问题是REPL环境特有的吗？
  不是，生成器的运行逻辑是Python语法层面的规则，在REPL环境、脚本运行、IDE执行等所有场景下表现完全一致，你在REPL里直接调用生成器函数只会输出生成器对象的内存地址，不会执行内部代码，和其他运行环境没有区别。

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