Python编程中实现迭代器的一些技巧小结


Posted in Python onJune 21, 2016

yield实现迭代器
如引言中的描述,实现一个可迭代的功能要是每次都手动实现iter,next稍稍有点麻烦,所需的代码也是比较客观。在python中也能通过借助yield的方式来实现一个迭代器。yield有一个关键的作能,它能够中断当前的执行逻辑,保持住现场(各种值的状态,执行的位置等等),返回相应的值,下一次执行的时候能够无缝的接着上次的地方继续执行,如此循环反复知道满足事先设置的退出条件或者发生错误强制被中断。
其具体功能是可以当return使用,从函数里返回一个值,不同之处是用yield返回之后,可以让函数从上回yield返回的地点继续执行。也就是说,yield返回函数,交给调用者一个返回值,然后再“瞬移”回去,让函数继续运行, 直到吓一跳yield语句再返回一个新的值。使用yield返回后,调用者实际得到的是一个迭代器对象,迭代器的值就是返回值,而调用该迭代器的next()方法会导致该函数恢复yield语句的执行环境继续往下跑,直到遇到下一个yield为止,如果遇不到yield,就会抛出异常表示迭代结束。
看一个例子:

>>> def test_yield():
... yield 1
... yield 2
... yield (1,2)
...
>>> a = test_yield()
>>> a.next()
1
>>> a.next()
2
>>> a.next()
(1, 2)
>>> a.next()
Traceback (most recent call last):
 File "<stdin>", line 1, in ?
StopIteration

光听描述就觉得和迭代器的工作方式很一致是吧,的确,yield能把它所在的函索变成一个迭代器,拿最经典的菲波那切数列的例子聊简述一下工作的方式:

def fab(max): 
 n, a, b = 0, 0, 1 
 while n < max:
 print b, "is generated" 
 yield b
 a, b = b, a + b 
 n = n + 1 

>>> for item in fab(5):
... print item
... 
1 is generated
1
1 is generated
1
2 is generated
2
3 is generated
3
5 is generated
5

我们有回想一下for关键字的语法糖,在这里遍历5以内的菲波那切数列值的时候,很显然fab(5)生成了一个可迭代的对象,遍历开始的时候它的iter方法被调用返回一个实际工作的迭代器对象,然后每一次调用它的next方法返回一个菲波那切数列值然后打印出来。
我们可以将调用生成器函数返回的对象的属性打印出来,看一下到底发生了什么:

>>> temp_gen = fab(5)
>>> dir(temp_gen)
['__class__', '__delattr__', '__doc__', '__format__', '__getattribute__', '__hash__', '__init__', '__iter__', '__name__', '__new__', '__reduce__', '__reduce_ex__', '__repr__', '__setattr__', '__sizeof__', '__str__', '__subclasshook__', 'close', 'gi_code', 'gi_frame', 'gi_running', 'next', 'send', 'throw']

正如上面的描述,单纯调用fab并不会让函数立刻开始返回任何值,并且从打印出的fab(5)的属性列表能够看到,生成器函数返回的对象包含有__iter__,next的实现。与我们手动实现相比,使用yield很方便的就能够实现我们想要的功能,代码量缩减不少。

Generator Expression
python中另一种能更优雅生成迭代器对象的方式就是使用生成器表达式Generator expression,它和列表解析表达式有着非常相似的写法,仅仅是把中括号[]变成()而已,不过小小改变产生的实际效果确实大大的不一样:

>>> temp_gen = (x for x in range(5))
>>> temp_gen
<generator object <genexpr> at 0x7192d8>
>>> for item in temp_gen:
... print item
... 
0
1
2
3
4
>>> dir(temp_gen)
['__class__', '__delattr__', '__doc__', '__format__', '__getattribute__', '__hash__', '__init__', '__iter__', '__name__', '__new__', '__reduce__', '__reduce_ex__', '__repr__', '__setattr__', '__sizeof__', '__str__', '__subclasshook__', 'close', 'gi_code', 'gi_frame', 'gi_running', 'next', 'send', 'throw']

看过上面对yield的描述,这个例子以及对应的输出日志还是相当直接明了的,无论是temp_gen的打印日志描述,for语句遍历的输出结果还是调用dir输出的属性列表,都赤裸裸的表明生成器表达式确实生成了能够支持迭代的对象。另外表达式里面也能够调用函数,增加适量的过滤条件。

内置库itertools 和 iter
python内置的库itertools提供了大量的工具方法,这些方法能够帮助我们创建能进行高效遍历和迭代的对象,里面包含不少有意思并且有用的方法,比如像chain, izip/izip_longest, combinations, ifilter等等。在python中还有一个内置的iter函数非常有用,能够返回一个迭代器对象,之后也就能够进行可以查看对应的帮助文档简单看一下:

>>> iter('abc')
<iterator object at 0x718590>
>>> str_iterator = iter('abc')
>>> next(str_iterator)
'a'
>>> next(str_iterator)
'b'
>>> lst_gen = iter([1,2,3,4])
>>> lst_gen
<listiterator object at 0x728e30>
>>> dir(lst_gen)
['__class__', '__delattr__', '__doc__', '__format__', '__getattribute__', '__hash__', '__init__', '__iter__', '__length_hint__', '__new__', '__reduce__', '__reduce_ex__', '__repr__', '__setattr__', '__sizeof__', '__str__', '__subclasshook__', 'next']

>>> help(iter)
Help on built-in function iter in module builtins:

iter(...)
 iter(iterable) -> iterator
 iter(callable, sentinel) -> iterator

 Get an iterator from an object. In the first form, the argument must
 supply its own iterator, or be a sequence.
 In the second form, the callable is called until it returns the sentinel.
Python 相关文章推荐
Python cookbook(数据结构与算法)对切片命名清除索引的方法
Mar 13 Python
Python实现按照指定要求逆序输出一个数字的方法
Apr 19 Python
TensorFlow实现模型评估
Sep 07 Python
浅析Python 3 字符串中的 STR 和 Bytes 有什么区别
Oct 14 Python
【python】matplotlib动态显示详解
Apr 11 Python
解决Pyinstaller 打包exe文件 取消dos窗口(黑框框)的问题
Jun 21 Python
django多个APP的urls设置方法(views重复问题解决)
Jul 19 Python
python 多维高斯分布数据生成方式
Dec 09 Python
python实现马丁策略的实例详解
Jan 15 Python
PyTorch 如何设置随机数种子使结果可复现
May 12 Python
python 开心网和豆瓣日记爬取的小爬虫
May 29 Python
python迷宫问题深度优先遍历实例
Jun 20 Python
Centos Python2 升级到Python3的简单实现
Jun 21 #Python
Python的Django框架中forms表单类的使用方法详解
Jun 21 #Python
Python正则表达式使用经典实例
Jun 21 #Python
常见的python正则用法实例讲解
Jun 21 #Python
实例讲解Python的函数闭包使用中应注意的问题
Jun 20 #Python
Python中的数学运算操作符使用进阶
Jun 20 #Python
Python中在for循环中嵌套使用if和else语句的技巧
Jun 20 #Python
You might like
日本十大最佳动漫,全都是二次元的神级作品
2019/10/05 日漫
php 中文字符串首字母的获取函数分享
2013/11/04 PHP
php堆排序实现原理与应用方法
2015/01/03 PHP
php 截取中英文混合字符串的方法
2018/05/31 PHP
使用jquery动态加载javascript以减少服务器压力
2012/10/29 Javascript
js 回车提交表单两种实现方法
2012/12/31 Javascript
jquery中post方法用法实例
2014/10/21 Javascript
推荐一个自己用的封装好的javascript插件
2015/01/29 Javascript
JQuery控制图片由中心点逐渐放大效果
2016/06/26 Javascript
javascript中sort排序实例详解
2016/07/24 Javascript
AngularJS的Filter的示例详解
2017/03/07 Javascript
ES6 更易于继承的类语法的使用
2019/02/11 Javascript
微信小程序实现的自定义分享功能示例
2019/02/12 Javascript
electron-vue利用webpack打包实现多页面的入口文件问题
2019/05/12 Javascript
一文搞懂ES6中的Map和Set
2019/05/20 Javascript
JavaScript直接调用函数与call调用的区别实例分析
2020/05/22 Javascript
vue 实现锚点功能操作
2020/08/10 Javascript
Javascript新手入门之字符串拼接与变量的应用
2020/12/03 Javascript
Python中的类与对象之描述符详解
2015/03/27 Python
Python实现的HTTP并发测试完整示例
2020/04/23 Python
使用Python3编写抓取网页和只抓网页图片的脚本
2015/08/20 Python
Python正则表达式如何进行字符串替换实例
2016/12/28 Python
Python3学习urllib的使用方法示例
2017/11/29 Python
Django中的Signal代码详解
2018/02/05 Python
python3 拼接字符串的7种方法
2018/09/12 Python
详解Appium+Python之生成html测试报告
2019/01/04 Python
python安装scipy的步骤解析
2019/09/28 Python
Python包和模块的分发详细介绍
2020/06/19 Python
linux centos 7.x 安装 python3.x 替换 python2.x的过程解析
2020/12/14 Python
全球领先的各类汽车配件零售商:Advance Auto Parts
2016/08/26 全球购物
捷克玩具商店:Bambule
2019/02/23 全球购物
销售工作岗位职责
2013/12/24 职场文书
建筑文秘专业个人求职信范文
2013/12/28 职场文书
公务员综合考察材料
2014/02/01 职场文书
2014年医院科室工作总结
2014/12/20 职场文书
CSS 实现角标效果的完整代码
2022/06/28 HTML / CSS