讲师中心 微信公众号
AI工具推荐 视频效率加速

Python 迭代器协议的完整实现方式

夏枫大大_9521

夏枫大大_9521

发布时间:2026-01-28 16:56:02

|

838人浏览过

|

来源于php中文网

原创

完整迭代器必须同时实现__iter__和__next__:__iter__需返回含__next__的对象(通常为self),__next__耗尽时须显式抛StopIteration;状态应存于实例属性而非类属性;支持多次迭代需每次__iter__返回新实例;生成器函数可更安全地替代手动实现。

python 迭代器协议的完整实现方式

__iter__ 和 __next__ 必须同时存在才能算完整迭代器

只实现 __iter__ 返回自身,却不定义 __next__,对象无法被 next() 调用,会抛出 TypeError: iterator does not have a __next__() method。反之,仅实现 __next__ 而没实现 __iter__,则无法用于 for 循环或 list() 等消费迭代器的场景,因为这些操作内部先调用 iter(),而 iter(obj) 要求对象必须有 __iter__ 方法(且返回一个迭代器)。

完整实现的关键是:

  • __iter__ 必须返回一个实现了 __next__ 的对象(通常是 self)
  • __next__ 必须在耗尽时显式抛出 StopIteration,不能靠返回 None 或静默退出
  • 多次调用 iter() 应该能获得独立的迭代状态(除非你刻意设计为单次可迭代)

手动管理状态时容易忽略的边界情况

比如实现一个从 0 数到 n 的计数器迭代器,常见错误是把索引初始化写在类属性里,导致所有实例共享状态:

class BadCounter:
    i = 0  # ❌ 类变量,所有实例共用
    def __init__(self, n):
        self.n = n
    def __iter__(self):
        return self
    def __next__(self):
        if self.i >= self.n:
            raise StopIteration
        val = self.i
        self.i += 1
        return val

正确做法是把状态放在实例属性中:

class GoodCounter:
    def __init__(self, n):
        self.n = n
        self.i = 0  # ✅ 实例变量
    def __iter__(self):
        return self
    def __next__(self):
        if self.i >= self.n:
            raise StopIteration
        val = self.i
        self.i += 1
        return val

另一个易错点:__next__ 中未处理 self.i 超出范围的瞬间(比如 n=0 时首次调用就该抛异常),会导致无限循环或越界访问。

立即学习“Python免费学习笔记(深入)”;

支持多次迭代需返回新迭代器实例

如果希望同一个对象能被多次 for 遍历(比如 list、range 的行为),就不能让 __iter__ 总是返回 self。否则第二次遍历时,内部状态已耗尽,直接抛 StopIteration。

python-parallelization
python-parallelization

将顺序 Python 代码转换为并行/并发实现,用于在需要并行化 Python 代码、提升性能时使用。

下载

解决方式是让 __iter__ 每次都新建一个干净的迭代器:

class RepeatableRange:
    def __init__(self, stop):
        self.stop = stop
    def __iter__(self):
        return RepeatableRangeIterator(self.stop)  # ✅ 新实例
<p>class RepeatableRangeIterator:
def <strong>init</strong>(self, stop):
self.stop = stop
self.i = 0
def <strong>next</strong>(self):
if self.i >= self.stop:
raise StopIteration
val = self.i
self.i += 1
return val
def <strong>iter</strong>(self):
return self

注意:RepeatableRangeIterator 自身也要实现 __iter__(返回自身),否则 iter(it) 会失败。

用生成器函数替代手动实现更安全

手动维护 __iter__ + __next__ 容易出状态错乱、漏抛 StopIteration、忘记重置等问题。Python 提供了更简洁可靠的替代:在 __iter__ 中返回一个生成器。

例如等价于上面 GoodCounter 的写法:

class GeneratorBasedCounter:
    def __init__(self, n):
        self.n = n
    def __iter__(self):
        i = 0
        while i < self.n:
            yield i
            i += 1

这种写法天然支持多次迭代(每次调用 __iter__ 都启一个新的生成器)、自动处理 StopIteration、无需手动管理状态变量生命周期。唯一要注意的是:生成器函数不能被 next() 直接调用(得先 iter()),但这本来就是迭代器协议的要求。

真正需要手写协议的场景其实很少——多数时候是封装底层流、资源或需要精细控制暂停/恢复逻辑时才值得深入。其他情况,生成器函数或 yield 就够用了。

热门AI工具

更多
DeepSeek

DeepSeek是一款面向对话、写作、编程和推理场景的AI大模型工具。

UP简历
UP简历 Hot

一款AI办公效率工具,主要用于基于AI技术的免费在线简历制作工具,适合需要提升相关任务效率的用户。

豆包大模型

豆包大模型是一款由字节跳动推出的企业级大语言模型服务平台。

切问学术

切问学术是一款AI论文写作工具,复旦大学NLP团队推出的AI学术智能体。

PixTV
PixTV Hot

PixTV是一款面向AIGC内容创作的AI视频生成工具。

WorkBuddy

一款AI办公效率工具,主要用于腾讯云推出的AI原生桌面智能体工作台,适合需要提升相关任务效率的用户。

讯飞智作

讯飞智作是一款AI视频创作工具,AI文本配音工具,数字人课程、营销视频制作。

SkildArt
SkildArt Hot

SkildArt是一款AI文本写作工具,一站式 AI 视觉创作平台。

火山引擎

火山引擎是一款面向企业的云计算与AI服务平台。

相关专题

更多
PixTV官网入口地址合集
PixTV官网入口地址合集

本专题汇总了 PixTV AI 一站式视频创作平台的官方入口与使用教程。无需下载软件,浏览器直接访问即可使用。平台将剧本、图像、视频、声音与剪辑整合在“无限画布”中,接入 GPT Image 2.5、Seedance 2.5 等头部模型。本专题整理了从新建画布、角色锚定、分镜拆分到视频生成与导出的完整操作指南,助你快速上手 AI 短剧与漫剧创作。

40

2026.10.10

Kratos框架HTTP与gRPC服务开发教程
Kratos框架HTTP与gRPC服务开发教程

本专题围绕Kratos框架双协议服务开发,涵盖HTTP路由与处理器编写、参数获取、gRPC服务实现与客户端调用、metadata上下文传递、encoding编解码注册、统一响应封装、超时控制与流式响应实现方法。

20

2026.10.10

Kratos框架Protobuf接口定义与代码生成合集
Kratos框架Protobuf接口定义与代码生成合集

本专题讲解Kratos框架接口定义体系,涵盖proto编写规范、proto add/client/server生成命令、http注解路由、validate校验、OpenAPI文档生成、跨服务proto复用与兼容性设计。

0

2026.10.10

C++虚函数怎么定义和调用
C++虚函数怎么定义和调用

C++虚函数是实现运行时多态的重要机制。本专题从virtual关键字的基本用法入手,介绍基类与派生类之间的函数重写、基类指针调用派生类方法,以及动态绑定的执行过程,帮助初学者掌握虚函数的核心语法。

20

2026.10.10

C++类与对象的封装方法教程
C++类与对象的封装方法教程

C++封装是面向对象编程的核心特性之一,通过类将数据与操作数据的函数组织在一起,并利用访问权限控制外部访问。本专题介绍类的定义、成员变量、成员函数以及public、private和protected的使用方法,帮助初学者掌握封装的基本原理。

20

2026.10.10

C++构造函数定义与调用方法
C++构造函数定义与调用方法

C++构造函数用于初始化类对象,是面向对象编程的重要基础。本专题从构造函数的定义、声明和调用入手,介绍默认构造函数、带参数构造函数、拷贝构造函数及成员初始化列表,帮助初学者掌握对象创建与初始化的基本方法。

20

2026.10.10

Kratos框架零基础入门教程
Kratos框架零基础入门教程

本专题整理Kratos框架入门内容,涵盖Go环境准备、kratos CLI安装升级、new命令创建项目、目录结构分层说明、服务启动与双协议端口、依赖下载报错排查,帮助开发者快速跑通第一个Kratos框架微服务应用。

20

2026.10.10

C++条件判断语句怎么写
C++条件判断语句怎么写

C++条件判断是控制程序执行流程的重要基础。本专题介绍if、if-else、else if和switch等常见分支语句,结合条件表达式、比较运算符与代码示例,帮助初学者掌握不同场景下的判断逻辑。

0

2026.10.10

C++变量怎么声明和赋值
C++变量怎么声明和赋值

C++变量是编写程序和存储数据的基础。本专题围绕变量声明、定义、初始化、赋值和类型选择等内容展开,帮助初学者理解不同变量的用法,并掌握在实际代码中定义和使用变量的方法。

20

2026.10.10

热门下载

更多
网站特效
/
网站源码
/
网站素材
/
前端模板

精品课程

更多
相关推荐
/
热门推荐
/
最新课程
关于我们 免责申明 举报中心 意见反馈 讲师合作 广告合作 最新更新
php中文网:公益在线php培训,帮助PHP学习者快速成长!
关注服务号
PHP中文网订阅号
每天精选资源文章推送

Copyright 2014-2026 https://www.php.cn/ All Rights Reserved | php.cn