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根据开头和结尾字符串获取中间字符串的方法
Mar 26 Python
在Django中限制已登录用户的访问的方法
Jul 23 Python
Python 爬虫学习笔记之单线程爬虫
Sep 21 Python
python通过微信发送邮件实现电脑关机
Jun 20 Python
对python PLT中的image和skimage处理图片方法详解
Jan 10 Python
使用Python向C语言的链接库传递数组、结构体、指针类型的数据
Jan 29 Python
Python中将两个或多个list合成一个list的方法小结
May 12 Python
pyqt5移动鼠标显示坐标的方法
Jun 21 Python
python如何实现从视频中提取每秒图片
Oct 22 Python
基于django ManyToMany 使用的注意事项详解
Aug 09 Python
pycharm中导入模块错误时提示Try to run this command from the system terminal
Mar 26 Python
JupyterNotebook 输出窗口的显示效果调整方法
Apr 13 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
PHP的FTP学习(二)[转自奥索]
2006/10/09 PHP
提取HTML标签
2006/10/09 PHP
PH P5.2至5.5、5.6的新增功能详解
2014/07/14 PHP
php实现TCP端口检测的方法
2015/04/01 PHP
盘点PHP和ASP.NET的10大对比!
2015/12/24 PHP
深入认识JavaScript中的函数
2007/01/22 Javascript
JavaScript 事件属性绑定带参数的函数
2009/03/13 Javascript
javascript日期转换 时间戳转日期格式
2011/11/05 Javascript
jQuery实现form表单reset按钮重置清空表单功能
2012/12/18 Javascript
js Object2String方便查看js对象内容
2014/11/24 Javascript
jQuery实现鼠标划过修改样式的方法
2015/04/14 Javascript
javascript显示中文日期的方法
2015/06/18 Javascript
轻松掌握jQuery中wrap()与unwrap()函数的用法
2016/05/24 Javascript
微信小程序 登陆流程详细介绍
2017/01/17 Javascript
javascript九宫格图片随机打乱位置的实现方法
2017/03/15 Javascript
JavaScript箭头(arrow)函数详解
2017/06/04 Javascript
vue 子组件向父组件传值方法
2018/02/26 Javascript
JavaScript遍历数组的三种方法map、forEach与filter实例详解
2019/02/27 Javascript
VUE组件中的 Drawer 抽屉实现代码
2019/08/06 Javascript
js实现多图和单图上传显示
2019/12/18 Javascript
jQuery 常用特效实例小结【显示与隐藏、淡入淡出、滑动、动画等】
2020/05/19 jQuery
Python入门篇之字符串
2014/10/17 Python
Python中read()、readline()和readlines()三者间的区别和用法
2017/07/30 Python
Flask框架各种常见装饰器示例
2018/07/17 Python
12个步骤教你理解Python装饰器
2019/07/01 Python
python挖矿算力测试程序详解
2019/07/03 Python
关于阿里云oss获取sts凭证 app直传 python的实例
2019/08/20 Python
Python 脚本拉取 Docker 镜像问题
2019/11/10 Python
python 实现检验33品种数据是否是正态分布
2019/12/09 Python
python实现移动木板小游戏
2020/10/09 Python
matplotlib 使用 plt.savefig() 输出图片去除旁边的空白区域
2021/01/05 Python
5分钟弄清楚html5的drag and drop(小结)
2019/04/10 HTML / CSS
linux面试题参考答案(5)
2016/11/05 面试题
会计与审计毕业生自荐信范文
2013/12/30 职场文书
秋季红领巾广播稿
2014/01/27 职场文书
python模板入门教程之flask Jinja
2022/04/11 Python