Python迭代器需实现__iter__()返回self和__next__()返回值或抛StopIteration;如Countdown类;生成器函数用yield自动实现迭代器。

Python 迭代器通过实现 __iter__() 和 __next__() 两个特殊方法来实现。对象只要具备这两个方法,就能被 for 循环、next() 等迭代工具使用。
迭代器的核心协议
Python 的迭代协议规定:一个对象要是迭代器,必须同时满足:
- 有 __iter__() 方法,且该方法返回对象本身(即
return self); - 有 __next__() 方法,每次调用返回下一个值,遍历完后抛出 StopIteration 异常。
手动实现一个简单迭代器
比如实现一个从 0 数到 n-1 的计数器迭代器:
class Countdown:
def __init__(self, n):
self.n = n
self.current = 0
def __iter__(self):
return self # 迭代器自身就是可迭代对象
def __next__(self):
if self.current >= self.n:
raise StopIteration
value = self.current
self.current += 1
return value使用示例
for i in Countdown(3):
print(i) # 输出:0, 1, 2
生成器函数:更简洁的迭代器写法
用 yield 定义的函数会自动返回一个迭代器对象,无需手动写 __iter__ 和 __next__:
立即学习“Python免费学习笔记(深入)”;
def countdown_gen(n):
current = 0
while current < n:
yield current
current += 1
效果等价于上面的 Countdown 类
for i in countdown_gen(3):
print(i) # 同样输出:0, 1, 2
生成器函数在首次调用时返回一个生成器对象(本质是迭代器),每次 next() 调用会执行到下一个 yield 并暂停,状态自动保存。
注意区分:可迭代对象 vs 迭代器
不是所有能用 for 遍历的对象都是迭代器:
-
可迭代对象(如 list、str、dict)只实现了
__iter__(),返回一个迭代器;
-
迭代器 是可迭代对象的“一次性的遍历工具”,既可被遍历,也实现了
__next__();
- 调用
iter(obj) 可把可迭代对象转为迭代器;调用 next(it) 触发实际取值。
基本上就这些。掌握协议本质比死记语法更重要——只要守住 __iter__ 返回自己、__next__ 按需返回值+抛异常这两条,就能写出任意逻辑的迭代器。











