yield from与for循环yield行为差异背后的机制是什么？

我原本以为yield from的作用就是逐个yield可迭代对象中的元素，但实际测试发现了不同行为。

我定义了两个类：

Class1（使用普通for循环+yield） #

class Class1:
    def __init__(self, gen):
        self.gen = gen
        
    def __iter__(self):
        for el in self.gen:
            yield el

Class2（使用yield from） #

class Class2:
    def __init__(self, gen):
        self.gen = gen
        
    def __iter__(self):
        yield from self.gen

执行以下测试代码：

a = Class1((i for i in range(3)))
print(next(iter(a)))
for el in iter(a):
    print(el)

我观察到两者输出结果不同：

• Class1输出：

0
1
2

• Class2输出：

0

请问导致这种差异的yield from背后的机制是什么？

首先要纠正一个点：正常情况下，你给出的Class2代码运行后，输出应该和Class1完全一致。如果观察到Class2只输出0，大概率是对测试代码的执行逻辑有误解，我们先拆解两者的工作机制：

1. Class1的迭代逻辑 #

Class1.__iter__是个生成器函数，每次调用iter(a)都会返回一个全新的生成器实例：

• 第一次调用next(iter(a))：生成器启动，通过for循环调用next(self.gen)拿到0，随后生成器暂停；此时传入的生成器self.gen状态停在0之后，并没有被耗尽。
• 第二次调用iter(a)：又生成一个新的生成器实例，这个生成器会继续从self.gen的当前状态（0之后）遍历，拿到1和2，所以输出完整的0、1、2。

2. yield from 的本质：迭代器委托 #

yield from self.gen的基础作用确实等价于for el in self.gen: yield el，但它是更底层的迭代器委托机制，和普通for+yield的核心差异在于：

• yield from会让当前生成器和被委托的迭代器（self.gen）建立更紧密的绑定，包括状态同步、双向值传递、异常转发等高级特性；
• 但在你的测试场景中，只要self.gen是状态可保留的生成器（比如你用的生成器表达式），两次调用iter(a)的逻辑和Class1完全一致：每次都会生成新的委托迭代器，继续消耗self.gen的剩余元素，输出应该也是0、1、2。

为什么会出现Class2只输出0的情况？ #

如果确实观察到Class2仅输出0，唯一的可能是两次调用iter(a)时，self.gen已经被完全耗尽，比如：

• 你在第一次调用next(iter(a))后，有其他代码隐式遍历完了self.gen；
• 或者你误将iter(a)的返回值复用（比如把it = iter(a)放在外面，先next(it)再for el in it）——但这种情况Class1和Class2的输出会完全一致，都能拿到剩余元素。

总结 #

你对yield from的基础认知没错，它核心作用就是逐个产出可迭代对象的元素。你观察到的差异大概率是测试场景中的误解导致的，而非yield from的基础机制问题。

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