python迭代器如何实现

舞夢輝影

发布： 2025-12-22 23:50:01

原创

196人浏览过

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

python迭代器如何实现

Python 迭代器通过实现 __iter__() 和 __next__() 两个特殊方法来实现。对象只要具备这两个方法，就能被 for 循环、next() 等迭代工具使用。

迭代器的核心协议

Python 的迭代协议规定：一个对象要是迭代器，必须同时满足：

有 __iter__() 方法，且该方法返回对象本身（即 return self）；
有 __next__() 方法，每次调用返回下一个值，遍历完后抛出 StopIteration 异常。

手动实现一个简单迭代器

比如实现一个从 0 数到 n-1 的计数器迭代器：

<font size="2"><pre class="brush:php;toolbar:false;">class Countdown:
    def __init__(self, n):
        self.n = n
        self.current = 0
<pre class='brush:python;toolbar:false;'>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免费学习笔记（深入）”；

易企CMS1.8

易企CMS：国内首款完全基于SEO友好性开发的营销型企业网站系统，让企业网络营销从此易如反掌。本程序特征：100%开发源代码，免费开源;后台管理操作简单易行;模板div+css标准设计,符合w3c标准，兼容主流浏览器;开发语言和数据库：PHP+Mysql。本程序亮点：从基础代码开发起完全符合SEOWHY理论的SEO规范，力图实现国内首款对SEO最友好的企业网站开源程序，为企业网络营销的巨大成功

查看详情

<font size="2"><pre class="brush:php;toolbar:false;">def countdown_gen(n):
    current = 0
    while current < n:
        yield current
        current += 1
<h1>效果等价于上面的 Countdown 类</h1><p>for i in countdown_gen(3):
print(i)  # 同样输出：0, 1, 2

登录后复制

生成器函数在首次调用时返回一个生成器对象（本质是迭代器），每次 next() 调用会执行到下一个 yield 并暂停，状态自动保存。