0

0

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

冰川箭仙

冰川箭仙

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

|

671人浏览过

|

来源于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免费学习笔记(深入)”;

典型结构:

Magician
Magician

Figma插件,AI生成图标、图片和UX文案

下载
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助手

腾讯元宝
腾讯元宝

腾讯混元平台推出的AI助手

文心一言
文心一言

文心一言是百度开发的AI聊天机器人,通过对话可以生成各种形式的内容。

讯飞写作
讯飞写作

基于讯飞星火大模型的AI写作工具,可以快速生成新闻稿件、品宣文案、工作总结、心得体会等各种文文稿

即梦AI
即梦AI

一站式AI创作平台,免费AI图片和视频生成。

ChatGPT
ChatGPT

最最强大的AI聊天机器人程序,ChatGPT不单是聊天机器人,还能进行撰写邮件、视频脚本、文案、翻译、代码等任务。

相关专题

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

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

186

2023.09.27

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

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

137

2025.07.29

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

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

137

2025.07.29

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

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

358

2023.06.29

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

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

2082

2023.08.14

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

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

349

2023.08.31

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

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

256

2023.09.05

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

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

326

2023.10.09

clawdbot ai使用教程 保姆级clawdbot部署安装手册
clawdbot ai使用教程 保姆级clawdbot部署安装手册

Clawdbot是一个“有灵魂”的AI助手,可以帮用户清空收件箱、发送电子邮件、管理日历、办理航班值机等等,并且可以接入用户常用的任何聊天APP,所有的操作均可通过WhatsApp、Telegram等平台完成,用户只需通过对话,就能操控设备自动执行各类任务。

19

2026.01.29

热门下载

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

精品课程

更多
相关推荐
/
热门推荐
/
最新课程
最新Python教程 从入门到精通
最新Python教程 从入门到精通

共4课时 | 22.4万人学习

Django 教程
Django 教程

共28课时 | 3.6万人学习

SciPy 教程
SciPy 教程

共10课时 | 1.3万人学习

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

Copyright 2014-2026 https://www.php.cn/ All Rights Reserved | php.cn | 湘ICP备2023035733号