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

Python中如何使用迭代器 迭代协议与自定义迭代器的实现

陌丽大大_1395

陌丽大大_1395

发布时间:2025-06-30 17:12:02

|

353人浏览过

|

来源于php中文网

原创

迭代器是一种用于遍历可迭代对象的工具,它提供统一的方式来访问集合元素而无需了解底层实现。1. 可迭代对象是定义了__iter__()或__getitem__()方法的对象;2. 迭代器对象同时实现了__iter__()和__next__()方法;3. iter()函数用于获取迭代器;4. next()函数用于获取下一个元素;5. 自定义迭代器需定义类并实现__iter__()和__next__()方法;6. 生成器使用yield关键字简化迭代器创建;7. 迭代器与可迭代对象的区别在于迭代器执行实际迭代;8. 可使用itertools模块处理无限序列并结合islice等函数控制循环。

Python中如何使用迭代器 迭代协议与自定义迭代器的实现

迭代器是一种让你能够遍历任何可迭代对象(比如列表、元组、字典、集合,甚至自定义的数据结构)的强大工具。它提供了一种统一的方式来访问集合中的元素,而无需了解底层数据结构的具体实现。简单来说,迭代器就是实现了特定协议的对象,允许你逐个访问数据,就像翻书一样,一次翻一页。

Python中如何使用迭代器 迭代协议与自定义迭代器的实现

解决方案:

Python中如何使用迭代器 迭代协议与自定义迭代器的实现

Python 中使用迭代器主要涉及以下几个关键点:

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

  1. 可迭代对象 (Iterable):一个对象,如果它定义了 __iter__() 方法,或者定义了 __getitem__() 方法(从 0 开始索引),那么它就是一个可迭代对象。简单说,就是可以用 for 循环遍历的对象。

    Python中如何使用迭代器 迭代协议与自定义迭代器的实现
  2. 迭代器对象 (Iterator):一个对象,它既实现了 __iter__() 方法(返回迭代器自身),又实现了 __next__() 方法(返回下一个值,如果没有值了就抛出 StopIteration 异常)。

  3. iter() 函数:用于从一个可迭代对象中获取迭代器。

  4. next() 函数:用于从迭代器中获取下一个元素。

简单示例:

my_list = [1, 2, 3]
my_iterator = iter(my_list) # 获取迭代器
print(next(my_iterator)) # 输出 1
print(next(my_iterator)) # 输出 2
print(next(my_iterator)) # 输出 3
# print(next(my_iterator)) # 抛出 StopIteration 异常

自定义迭代器:

Shadows Python Sensei
Shadows Python Sensei

Python 最佳实践助手——代码规范、设计模式、性能优化、测试与类型注解。适用于编写或审查 Python 代码。

下载

要创建自定义迭代器,你需要定义一个类,并实现 __iter__()__next__() 方法。

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

    def __iter__(self):
        return self

    def __next__(self):
        if self.index < len(self.data):
            value = self.data[self.index]
            self.index += 1
            return value
        else:
            raise StopIteration

# 使用自定义迭代器
my_list = [4, 5, 6]
my_iter = MyIterator(my_list)
for item in my_iter:
    print(item) # 输出 4, 5, 6

迭代协议:

迭代协议定义了迭代器应该如何工作。它包含两个关键方法:

  • __iter__(): 返回迭代器对象本身。这允许在需要迭代器的地方使用对象本身。
  • __next__(): 返回序列中的下一个项目。如果在迭代器中没有更多的项目,则引发 StopIteration 异常。

为什么使用迭代器而不是直接索引?

迭代器提供了一种更通用的访问元素的方式。 想象一下,你有一个非常大的文件,如果一次性加载到内存中,可能会导致内存溢出。 迭代器允许你逐行读取文件,处理完一行再读取下一行,而无需一次性加载整个文件。 这对于处理大数据集或无限序列非常有用。另外,迭代器可以隐藏底层数据结构的复杂性,提供一个更简洁的接口。

如何使用生成器创建迭代器?

生成器是一种特殊的迭代器,它使用 yield 关键字来生成值。 使用生成器可以更简洁地创建迭代器,而无需显式定义 __iter__()__next__() 方法。

def my_generator(data):
    for item in data:
        yield item * 2

# 使用生成器
my_list = [7, 8, 9]
my_gen = my_generator(my_list)
for item in my_gen:
    print(item) # 输出 14, 16, 18

生成器函数在遇到 yield 关键字时会暂停执行,并将 yield 后面的值返回。 当再次调用 next() 函数时,生成器函数会从上次暂停的地方继续执行,直到遇到下一个 yield 关键字或函数结束。 这种方式使得生成器可以按需生成值,节省内存。

迭代器和可迭代对象有什么区别?

可迭代对象是可以返回迭代器的对象。 换句话说,可迭代对象实现了 __iter__() 方法,该方法返回一个迭代器。 迭代器是实际执行迭代的对象。 它实现了 __next__() 方法,该方法返回序列中的下一个元素。

一个常见的误解是认为列表本身就是迭代器。 实际上,列表是可迭代对象,但它不是迭代器。 你需要使用 iter() 函数从列表中获取迭代器。 迭代器只能使用一次,当它耗尽时,你需要从可迭代对象中获取一个新的迭代器。

如何处理无限序列的迭代?

迭代器非常适合处理无限序列,因为它们可以按需生成值。 你可以使用 itertools 模块中的函数来创建无限迭代器,例如 count()cycle()repeat()

import itertools

# 创建一个从 1 开始无限递增的迭代器
counter = itertools.count(1)
print(next(counter)) # 输出 1
print(next(counter)) # 输出 2
print(next(counter)) # 输出 3

# 创建一个无限循环列表的迭代器
colors = itertools.cycle(['red', 'green', 'blue'])
print(next(colors)) # 输出 red
print(next(colors)) # 输出 green
print(next(colors)) # 输出 blue
print(next(colors)) # 输出 red

# 创建一个重复字符串 "hello" 无限次的迭代器
repeater = itertools.repeat("hello")
print(next(repeater)) # 输出 hello
print(next(repeater)) # 输出 hello
print(next(repeater)) # 输出 hello

在使用无限迭代器时,务必小心,确保你的代码能够正确地终止迭代,否则可能会导致无限循环。 通常,你需要使用 break 语句或 itertools 模块中的 islice() 函数来限制迭代的次数。

热门AI工具

更多
WorkBuddy

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

火山引擎

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

UP简历
UP简历 Hot

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

墨刀AI
墨刀AI Hot

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

豆包大模型

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

Laper
Laper Hot

Laper是专为编剧、导演和制片人推出的 AI 原生剧本创作工具。

二狗PPT
二狗PPT Hot

一款AI演示文稿工具,主要用于专为中式职场打造的AI PPT生成工具,适合需要提升相关任务效率的用户。

UpDream
UpDream Hot

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

DeepSeek

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

相关专题

更多
counta和count的区别
counta和count的区别

Count函数用于计算指定范围内数字的个数,而CountA函数用于计算指定范围内非空单元格的个数。本专题为大家提供相关的文章、下载、课程内容,供大家免费下载体验。

2328

2023.11.20

java中break的作用
java中break的作用

本专题整合了java中break的用法教程,阅读专题下面的文章了解更多详细内容。

2100

2025.10.15

java break和continue
java break和continue

本专题整合了java break和continue的区别相关内容,阅读专题下面的文章了解更多详细内容。

647

2025.10.24

treenode的用法
treenode的用法

​在计算机编程领域,TreeNode是一种常见的数据结构,通常用于构建树形结构。在不同的编程语言中,TreeNode可能有不同的实现方式和用法,通常用于表示树的节点信息。更多关于treenode相关问题详情请看本专题下面的文章。php中文网欢迎大家前来学习。

2121

2023.12.01

C++ 高效算法与数据结构
C++ 高效算法与数据结构

本专题讲解 C++ 中常用算法与数据结构的实现与优化,涵盖排序算法(快速排序、归并排序)、查找算法、图算法、动态规划、贪心算法等,并结合实际案例分析如何选择最优算法来提高程序效率。通过深入理解数据结构(链表、树、堆、哈希表等),帮助开发者提升 在复杂应用中的算法设计与性能优化能力。

296

2025.12.22

深入理解算法:高效算法与数据结构专题
深入理解算法:高效算法与数据结构专题

本专题专注于算法与数据结构的核心概念,适合想深入理解并提升编程能力的开发者。专题内容包括常见数据结构的实现与应用,如数组、链表、栈、队列、哈希表、树、图等;以及高效的排序算法、搜索算法、动态规划等经典算法。通过详细的讲解与复杂度分析,帮助开发者不仅能熟练运用这些基础知识,还能在实际编程中优化性能,提高代码的执行效率。本专题适合准备面试的开发者,也适合希望提高算法思维的编程爱好者。

337

2026.01.06

C++ 数据结构与算法实现教程合集
C++ 数据结构与算法实现教程合集

以 C++ 为实现语言,系统讲解核心数据结构与算法,涵盖链表(单链表/双链表/环检测)、栈与队列(单调栈/优先队列)、二叉树(遍历/BST/AVL/红黑树)、哈希表(开地址法/链地址法)、图(邻接表/BFS/DFS/Dijkstra/拓扑排序)、常见排序算法(快排/归并/堆排/计数排序)的实现与复杂度分析,同时分享 LeetCode 刷题技巧、竞赛编程常用模板(二分/前缀和/滑动窗口/动态规划),帮助开发者夯实算法基础。

372

2026.05.09

硬盘接口类型介绍
硬盘接口类型介绍

硬盘接口类型有IDE、SATA、SCSI、Fibre Channel、USB、eSATA、mSATA、PCIe等等。详细介绍:1、IDE接口是一种并行接口,主要用于连接硬盘和光驱等设备,它主要有两种类型:ATA和ATAPI,IDE接口已经逐渐被SATA接口;2、SATA接口是一种串行接口,相较于IDE接口,它具有更高的传输速度、更低的功耗和更小的体积;3、SCSI接口等等。

3028

2023.10.19

Conan私有仓库搭建教程
Conan私有仓库搭建教程

本专题系统的讲解Conan私有仓库的搭建流程,涵盖仓库服务部署、存储目录配置、用户认证、权限划分和远程地址添加,并介绍内部C++依赖包的上传、下载及版本维护方法。

0

2026.09.22

热门下载

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

精品课程

更多
相关推荐
/
热门推荐
/
最新课程
ECMAScript6 / ES6---十天技能课堂
ECMAScript6 / ES6---十天技能课堂

共25课时 | 2.7万人学习

Conan 2 安装指南
Conan 2 安装指南

共0课时 | 0人学习

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

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