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

Python 自定义可迭代对象的正确姿势

冬枫酱_3669

冬枫酱_3669

发布时间:2026-01-29 18:46:36

|

691人浏览过

|

来源于php中文网

原创

__iter__ 必须返回迭代器而非数据,因Python迭代机制只认迭代器协议(含__next__且耗尽时抛StopIteration);直接返回列表等虽可迭代,但丧失控制权、无法干预next()行为、多次遍历状态共享。

python 自定义可迭代对象的正确姿势

为什么 __iter__ 必须返回迭代器,而不是直接返回数据?

因为 Python 的 for 循环、list()、next() 等机制只认「迭代器协议」:对象要有 __next__ 方法,且在耗尽时抛出 StopIteration。如果 __iter__ 直接返回列表或生成器表达式(比如 return self.data),而 self.data 是 list,那它确实可迭代——但这是靠 list 自己实现了迭代器协议;你的类只是“借壳”,没真正控制迭代逻辑。

正确做法是让 __iter__ 返回一个满足协议的对象:

  • 返回一个自定义迭代器类的实例(推荐,逻辑清晰、可维护)
  • 返回生成器函数(即含 yield 的函数)的调用结果(简洁,但状态封装弱)
  • 避免返回 iter(self.data) 除非你明确只需要代理行为且不扩展逻辑

示例中常见错误:def __iter__(self): return self.data —— 这会让多次遍历共享内部状态(比如 list 的索引不可控),也失去对 next() 行为的干预能力。

自定义迭代器类里,__next__ 怎么写才不出错?

核心就两点:有明确的终止条件 + 每次调用返回下一个值。最容易踩的坑是忘记维护游标、越界不抛异常、或在耗尽后继续返回值。

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

macOS UI 自动化技能包 —— 用 Python 控制你的 Mac
macOS UI 自动化技能包 —— 用 Python 控制你的 Mac

让 macOS 自动化像说话一样简单 —— 用代码控制你的 Mac,无需学习 AppleScript 系统要求 macOS 14+ (Sonoma) Python 3.9+ 需要开启:辅助功能 + 屏幕录制权限

下载

典型结构:

class MyIterator:
    def __init__(self, data):
        self.data = data
        self.index = 0

    def __next__(self):
        if self.index >= len(self.data):
            raise StopIteration
        value = self.data[self.index]
        self.index += 1
        return value
  • 必须显式检查边界,不能依赖 try/except IndexError —— 迭代器协议要求抛 StopIteration,不是 IndexError
  • 不要在 __next__ 里重置 self.index,否则会导致无限循环
  • 如果数据支持动态修改(如边迭代边增删),需额外考虑一致性,通常建议迭代期间禁止修改

用生成器函数实现 __iter__ 时,哪些细节会影响行为?

写成 def __iter__(self): yield from self.data 或手动 yield 是合法且常见的,但要注意:

  • 每次调用 __iter__ 都会创建新生成器,天然支持多次独立遍历(这点比返回同一迭代器实例更安全)
  • 生成器无法倒带或重复使用,符合迭代器语义,但没法像类迭代器那样暴露 .reset() 或 .peek() 等方法
  • 如果需要在迭代中访问外部状态(如计数、缓存、IO 控制),生成器函数的闭包变量不如类属性直观,容易引发意外共享
  • 调试困难:生成器对象没有公开的当前状态字段,print(gen) 看不到游标位置

所以简单代理用生成器,复杂逻辑(如分页、过滤、懒加载)优先用独立迭代器类。

为什么 __len__ 和 __getitem__ 不能替代 __iter__?

有些同学以为只要实现了 __getitem__(支持下标访问)和 __len__,Python 就会自动提供迭代——这没错,但它是「后备机制」:当类没定义 __iter__ 时,解释器会尝试用 __getitem__ 从 0 开始调用,直到抛出 IndexError。

  • 这种自动迭代无法中断或定制(比如跳过空项、提前退出)
  • 如果 __getitem__ 不是 O(1)(比如要查数据库),性能灾难
  • 一旦你加了 __iter__,后备机制就失效了,哪怕你写的是 pass —— 所以别依赖它
  • __len__ 完全无关迭代协议,只是方便 len() 调用;很多可迭代对象根本没法高效算长度(如文件行、网络流)

真正可控、可预测、可扩展的迭代,必须显式实现 __iter__,而且它的返回值必须严格遵循迭代器协议——这个契约比看起来更硬,绕不开。

热门AI工具

更多
豆包大模型

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

DeepSeek

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

VibeKnow
VibeKnow Hot

一款AI视频创作工具,主要用于全球首个AI知识视频创作平台,文档、文章、网页,一键生成视频,适合需要提升相关任务效率的用户。

讯飞智作

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

WorkBuddy

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

花生AI
花生AI Hot

一款AI视频创作工具,主要用于B站推出的AI视频创作工具,适合需要提升相关任务效率的用户。

AionClaw
AionClaw Hot

AionClaw是一款面向办公、创作和编程任务的AI桌面智能体。

墨刀AI
墨刀AI Hot

一款AI图像与设计工具,主要用于产品经理的专属智能体,适合需要提升相关任务效率的用户。

Loomy
Loomy Hot

一款AI工具,主要用于科大讯飞发布的桌面级 AI 助理,比 OpenClaw 更易用、更安全!,适合需要提升相关任务效率的用户。

相关专题

更多
python中print函数的用法
python中print函数的用法

python中print函数的语法是“print(value1, value2, ..., sep=' ', end=' ', file=sys.stdout, flush=False)”。本专题为大家提供print相关的文章、下载、课程内容,供大家免费下载体验。

3140

2023.09.27

python print用法与作用
python print用法与作用

本专题整合了python print的用法、作用、函数功能相关内容,阅读专题下面的文章了解更多详细教程。

269

2026.02.03

go语言闭包相关教程大全
go语言闭包相关教程大全

本专题整合了go语言闭包相关数据,阅读专题下面的文章了解更多相关内容。

4733

2025.07.29

数据库三范式
数据库三范式

数据库三范式是一种设计规范,用于规范化关系型数据库中的数据结构,它通过消除冗余数据、提高数据库性能和数据一致性,提供了一种有效的数据库设计方法。本专题提供数据库三范式相关的文章、下载和课程。

2925

2023.06.29

如何删除数据库
如何删除数据库

删除数据库是指在MySQL中完全移除一个数据库及其所包含的所有数据和结构,作用包括:1、释放存储空间;2、确保数据的安全性;3、提高数据库的整体性能,加速查询和操作的执行速度。尽管删除数据库具有一些好处,但在执行任何删除操作之前,务必谨慎操作,并备份重要的数据。删除数据库将永久性地删除所有相关数据和结构,无法回滚。

4081

2023.08.14

vb怎么连接数据库
vb怎么连接数据库

在VB中,连接数据库通常使用ADO(ActiveX 数据对象)或 DAO(Data Access Objects)这两个技术来实现:1、引入ADO库;2、创建ADO连接对象;3、配置连接字符串;4、打开连接;5、执行SQL语句;6、处理查询结果;7、关闭连接即可。

3271

2023.08.31

MySQL恢复数据库
MySQL恢复数据库

MySQL恢复数据库的方法有使用物理备份恢复、使用逻辑备份恢复、使用二进制日志恢复和使用数据库复制进行恢复等。本专题为大家提供MySQL数据库相关的文章、下载、课程内容,供大家免费下载体验。

1027

2023.09.05

vb中怎么连接access数据库
vb中怎么连接access数据库

vb中连接access数据库的步骤包括引用必要的命名空间、创建连接字符串、创建连接对象、打开连接、执行SQL语句和关闭连接。本专题为大家提供连接access数据库相关的文章、下载、课程内容,供大家免费下载体验。

2847

2023.10.09

Vite 项目如何打包和部署
Vite 项目如何打包和部署

Vite项目打包部署专题汇集生产构建、产物预览、输出目录、资源路径、Nginx与GitHub Pages部署及常见白屏和路由404问题的已审核文章,帮助读者完成上线流程。

60

2026.10.28

热门下载

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

精品课程

更多
相关推荐
/
热门推荐
/
最新课程
Uvicorn官方安装与入门手册
Uvicorn官方安装与入门手册

共0课时 | 0人学习

Python uv官方项目管理指南
Python uv官方项目管理指南

共0课时 | 0人学习

Python uv官方安装与入门文档
Python uv官方安装与入门文档

共0课时 | 0人学习

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

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