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 相关文章推荐
Django与遗留的数据库整合的方法指南
Jul 24 Python
Python制作钉钉加密/解密工具
Dec 07 Python
TensorFlow数据输入的方法示例
Jun 19 Python
python面向对象入门教程之从代码复用开始(一)
Dec 11 Python
Python实现蒙特卡洛算法小实验过程详解
Jul 12 Python
使用Pyinstaller转换.py文件为.exe可执行程序过程详解
Aug 06 Python
python爬虫爬取笔趣网小说网站过程图解
Nov 18 Python
使用python 对验证码图片进行降噪处理
Dec 18 Python
Python3 使用selenium插件爬取苏宁商家联系电话
Dec 23 Python
TensorFlow命名空间和TensorBoard图节点实例
Jan 23 Python
Python Flask框架实现简单加法工具过程解析
Jun 03 Python
Python调用百度OCR实现图片文字识别的示例代码
Jul 17 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中的Swoole
2016/04/05 PHP
使用PHP+MySql+Ajax+jQuery实现省市区三级联动功能示例
2017/09/15 PHP
php读取本地json文件的实例
2018/03/07 PHP
PHP设计模式之模板方法模式定义与用法详解
2018/04/02 PHP
event.keyCode键码值表 附只能输入特定的字符串代码
2009/05/15 Javascript
IE6 弹出Iframe层中的文本框“经常”无法获得输入焦点
2009/12/27 Javascript
javascript showModalDialog 内跳转页面的问题
2010/11/25 Javascript
JavaScript入门之事件、cookie、定时等
2011/10/21 Javascript
JS刷新当前页面的几种方法总结
2013/12/24 Javascript
Javascript表单验证要注意的事项
2014/09/29 Javascript
Javascript基础教程之变量
2015/01/18 Javascript
深入探讨javascript函数式编程
2015/10/11 Javascript
jQuery中数据缓存$.data的用法及源码完全解析
2016/04/29 Javascript
深入理解JS实现快速排序和去重
2016/10/17 Javascript
Angular2使用Angular-CLI快速搭建工程(二)
2017/05/21 Javascript
es7学习教程之Decorators(修饰器)详解
2017/07/21 Javascript
Vue2.0 实现移动端图片上传功能
2018/05/30 Javascript
从零开始学习搭建React脚手架项目
2018/08/23 Javascript
iview同时验证多个表单问题总结
2018/09/29 Javascript
[32:17]完美世界DOTA2联赛循环赛LBZS vs Forest第二场 10月30日
2020/10/31 DOTA
用Python编写一个国际象棋AI程序
2014/11/28 Python
详解Python中contextlib上下文管理模块的用法
2016/06/28 Python
Python爬取十篇新闻统计TF-IDF
2018/01/03 Python
数据清洗--DataFrame中的空值处理方法
2018/07/03 Python
Python利用lxml模块爬取豆瓣读书排行榜的方法与分析
2019/04/15 Python
Python PyInstaller安装和使用教程详解
2020/01/08 Python
python实现银行实战系统
2020/02/26 Python
Win10环境中如何实现python2和python3并存
2020/07/20 Python
python中使用asyncio实现异步IO实例分析
2021/02/26 Python
澳大利亚男士西服品牌:M.J.Bale
2018/02/06 全球购物
什么是用户模式(User Mode)与内核模式(Kernel Mode) ?
2015/09/07 面试题
元旦联欢会主持词
2014/03/26 职场文书
学前教育见习总结
2015/06/23 职场文书
教你怎么用Python生成九宫格照片
2021/05/20 Python
vue-cli4.5.x快速搭建项目
2021/05/30 Vue.js
Pandas实现DataFrame的简单运算、统计与排序
2022/03/31 Python