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算法演练_One Rule 算法(详解)
May 17 Python
python 中random模块的常用方法总结
Jul 08 Python
使用python编写简单的小程序编译成exe跑在win10上
Jan 15 Python
python实现朴素贝叶斯分类器
Mar 28 Python
Python使用一行代码获取上个月是几月
Aug 30 Python
django认证系统 Authentication使用详解
Jul 22 Python
python3.7环境下安装Anaconda的教程图解
Sep 10 Python
pygame编写音乐播放器的实现代码示例
Nov 19 Python
使用Python测试Ping主机IP和某端口是否开放的实例
Dec 17 Python
tensorflow 动态获取 BatchSzie 的大小实例
Jun 30 Python
详解java调用python的几种用法(看这篇就够了)
Dec 10 Python
Python PIL按比例裁剪图片
May 11 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 SQLite类
2009/05/07 PHP
flash用php连接数据库的代码
2011/04/21 PHP
Linux下PHP加速器APC的安装与配置笔记
2014/10/24 PHP
php面向对象之反射功能与用法分析
2017/03/29 PHP
PHP中$GLOBALS与global的区别详解
2019/03/21 PHP
js数据验证集合、js email验证、js url验证、js长度验证、js数字验证等简单封装
2010/05/15 Javascript
jQuery中:first选择器用法实例
2014/12/30 Javascript
jQuery中removeProp()方法用法实例
2015/01/05 Javascript
JavaScript编程中布尔对象的基本使用
2015/10/25 Javascript
JS实现的鼠标跟随代码(卡通手型点击效果)
2015/10/26 Javascript
浅谈javascript中的call、apply、bind
2016/03/06 Javascript
使用Bootstrap typeahead插件实现搜索框自动补全的方法
2016/07/07 Javascript
使用jquery.qrcode.js生成二维码插件
2016/10/17 Javascript
js学习总结_基于数据类型检测的四种方式(必看)
2017/07/04 Javascript
jQuery简单实现向列表动态添加新元素的方法示例
2017/12/25 jQuery
electron 如何将任意资源打包的方法步骤
2020/04/16 Javascript
Vue 构造选项 - 进阶使用说明
2020/08/14 Javascript
Python collections模块实例讲解
2014/04/07 Python
Python使用shelve模块实现简单数据存储的方法
2015/05/20 Python
Python实现Windows上气泡提醒效果的方法
2015/06/03 Python
python创建与遍历List二维列表的方法
2019/08/16 Python
python 检查数据中是否有缺失值,删除缺失值的方式
2019/12/02 Python
python利用百度云接口实现车牌识别的示例
2020/02/21 Python
浅谈python 中的 type(), dtype(), astype()的区别
2020/04/09 Python
利用python对excel中一列的时间数据更改格式操作
2020/07/14 Python
使用Python提取文本中含有特定字符串的方法示例
2020/12/09 Python
模范教师事迹材料
2014/02/10 职场文书
2014年母亲节寄语
2014/05/07 职场文书
个人查摆剖析材料
2014/10/16 职场文书
2015年三年级班主任工作总结
2015/05/21 职场文书
Golang二维切片初始化的实现
2021/04/08 Golang
解决redis sentinel 频繁主备切换的问题
2021/04/12 Redis
java泛型通配符详解
2021/07/25 Java/Android
spring cloud eureka 服务启动失败的原因分析及解决方法
2022/03/17 Java/Android
Pandas实现批量拆分与合并Excel的示例代码
2022/05/30 Python
Redis+AOP+自定义注解实现限流
2022/06/28 Redis