文章详情

短信预约-IT技能 免费直播动态提醒

请输入下面的图形验证码

提交验证

短信预约提醒成功

Python的迭代器协议

2023-01-31 02:54

关注

    迭代器是Python中的一个高级概念,迭代器是一个实现了迭代器协议的对象,那何为迭代器协议呢?

满足下面两个条件就行。(1)该对象实现了__iter__()方法;(2)该对象实现了next()方法,返回当前元素,并指向下一个元素的位置,如果当前位置已无元素,则抛出StopIteration异常。

看下面例子

wKioL1f0uGiATtRcAACLQZmHWiE997.png-wh_50

这说明列表这个对象实现了迭代器协议。如果一个对象有__iter__()方法,那么它就是可迭代对象。可以利用迭代器协议来遍历可迭代对象,例如下面例子:

#coding=utf-8
a = range(3)
it = a.__iter__()
while 1:
    try:
        print it.next()
    except StopIteration:
        break

输出结果:

1

2


其实吧,for语句就是获取了容器的迭代器、调用迭代器的next()方法以及对StopIteration进行处理等流程进行封装了的语法糖,类似的还有in/not in语句。

迭代器的优势有很多:

(1)定义了统一的访问容器的接口,我们可以随时定义自己的迭代器,只要实现了迭代器协议就行。

(2)迭代器有惰性求值的特性,它进可以在迭代至当前元素时才计算,所以非常适合遍历元素非常多的容器,而且更省内存。

看下面例子:

#coding=utf-8
class Fab(object):
    def __init__(self, max):
        self.max = max
        self.n, self.a, self.b = 0, 0, 1

    def __iter__(self):
        return self

    def next(self):
        if self.n < self.max:
            r = self.b
            self.a, self.b = self.b, self.a + self.b
            self.n = self.n + 1
            return r
        raise StopIteration()

for one in Fab(5):
    print one

输出结果如下:

1

2

3

4

5

迭代器使用的内存会更小。看看下面的例子:

print sys.getsizeof(Fab(5))
print sys.getsizeof([1,2,3,4,5])


输出结果如下:

wKiom1f0vn2jacRKAAAaMtP6CdQ048.png-wh_50

同样是获取含有元素1,2,3,4,5的容器,迭代器更省内存。还有一个,看下面例子:

print sys.getsizeof(Fab(5))
print sys.getsizeof(Fab(50))
print sys.getsizeof(Fab(500))
print sys.getsizeof(Fab(5000))

输出结果如下:

wKioL1f0v7iziCTIAAAZsyckEmc810.png-wh_50

占用的内存并没有因为元素的增多而增加,所以在编写代码时要多多使用迭代器或者迭代协议。


Python中内置了一个iter()函数,可以返回一个迭代器对象,它接受的参数是一个实现了__iter__()方法的容器(也就是可迭代对象)或者迭代器。对于有__iter__()方法的容器,__iter__()也返回一个迭代器对象。


python中有itertools模块,里面的函数都是用迭代器实现的,效率很高,有时间你可以去了解下。

阅读原文内容投诉

免责声明:

① 本站未注明“稿件来源”的信息均来自网络整理。其文字、图片和音视频稿件的所属权归原作者所有。本站收集整理出于非商业性的教育和科研之目的,并不意味着本站赞同其观点或证实其内容的真实性。仅作为临时的测试数据,供内部测试之用。本站并未授权任何人以任何方式主动获取本站任何信息。

② 本站未注明“稿件来源”的临时测试数据将在测试完成后最终做删除处理。有问题或投稿请发送至: 邮箱/279061341@qq.com QQ/279061341

软考中级精品资料免费领

  • 历年真题答案解析
  • 备考技巧名师总结
  • 高频考点精准押题
  • 2024年上半年信息系统项目管理师第二批次真题及答案解析(完整版)

    难度     813人已做
    查看
  • 【考后总结】2024年5月26日信息系统项目管理师第2批次考情分析

    难度     354人已做
    查看
  • 【考后总结】2024年5月25日信息系统项目管理师第1批次考情分析

    难度     318人已做
    查看
  • 2024年上半年软考高项第一、二批次真题考点汇总(完整版)

    难度     435人已做
    查看
  • 2024年上半年系统架构设计师考试综合知识真题

    难度     224人已做
    查看

相关文章

发现更多好内容

猜你喜欢

AI推送时光机
位置:首页-资讯-后端开发
咦!没有更多了?去看看其它编程学习网 内容吧
首页课程
资料下载
问答资讯