介绍Python中内置的itertools模块


Posted in Python onApril 29, 2015

Python的内建模块itertools提供了非常有用的用于操作迭代对象的函数。

首先,我们看看itertools提供的几个“无限”迭代器:

>>> import itertools
>>> natuals = itertools.count(1)
>>> for n in natuals:
...   print n
...
1
2
3
...

因为count()会创建一个无限的迭代器,所以上述代码会打印出自然数序列,根本停不下来,只能按Ctrl+C退出。

cycle()会把传入的一个序列无限重复下去:

>>> import itertools
>>> cs = itertools.cycle('ABC') # 注意字符串也是序列的一种
>>> for c in cs:
...   print c
...
'A'
'B'
'C'
'A'
'B'
'C'
...

同样停不下来。

repeat()负责把一个元素无限重复下去,不过如果提供第二个参数就可以限定重复次数:

>>> ns = itertools.repeat('A', 10)
>>> for n in ns:
...   print n
...

打印10次'A'

无限序列只有在for迭代时才会无限地迭代下去,如果只是创建了一个迭代对象,它不会事先把无限个元素生成出来,事实上也不可能在内存中创建无限多个元素。

无限序列虽然可以无限迭代下去,但是通常我们会通过takewhile()等函数根据条件判断来截取出一个有限的序列:

>>> natuals = itertools.count(1)
>>> ns = itertools.takewhile(lambda x: x <= 10, natuals)
>>> for n in ns:
...   print n
...

打印出1到10

itertools提供的几个迭代器操作函数更加有用:
chain()

chain()可以把一组迭代对象串联起来,形成一个更大的迭代器:

for c in chain('ABC', 'XYZ'):
  print c
# 迭代效果:'A' 'B' 'C' 'X' 'Y' 'Z'

groupby()

groupby()把迭代器中相邻的重复元素挑出来放在一起:

>>> for key, group in itertools.groupby('AAABBBCCAAA'):
...   print key, list(group) # 为什么这里要用list()函数呢?
...
A ['A', 'A', 'A']
B ['B', 'B', 'B']
C ['C', 'C']
A ['A', 'A', 'A']

实际上挑选规则是通过函数完成的,只要作用于函数的两个元素返回的值相等,这两个元素就被认为是在一组的,而函数返回值作为组的key。如果我们要忽略大小写分组,就可以让元素'A'和'a'都返回相同的key:

>>> for key, group in itertools.groupby('AaaBBbcCAAa', lambda c: c.upper()):
...   print key, list(group)
...
A ['A', 'a', 'a']
B ['B', 'B', 'b']
C ['c', 'C']
A ['A', 'A', 'a']

imap()

imap()和map()的区别在于,imap()可以作用于无穷序列,并且,如果两个序列的长度不一致,以短的那个为准。

>>> for x in itertools.imap(lambda x, y: x * y, [10, 20, 30], itertools.count(1)):
...   print x
...
10
40
90

注意imap()返回一个迭代对象,而map()返回list。当你调用map()时,已经计算完毕:

>>> r = map(lambda x: x*x, [1, 2, 3])
>>> r # r已经计算出来了
[1, 4, 9]

当你调用imap()时,并没有进行任何计算:

>>> r = itertools.imap(lambda x: x*x, [1, 2, 3])
>>> r
<itertools.imap object at 0x103d3ff90>
# r只是一个迭代对象

必须用for循环对r进行迭代,才会在每次循环过程中计算出下一个元素:

>>> for x in r:
...   print x
...
1
4
9

这说明imap()实现了“惰性计算”,也就是在需要获得结果的时候才计算。类似imap()这样能够实现惰性计算的函数就可以处理无限序列:

>>> r = itertools.imap(lambda x: x*x, itertools.count(1))
>>> for n in itertools.takewhile(lambda x: x<100, r):
...   print n
...

结果是什么?

如果把imap()换成map()去处理无限序列会有什么结果?

>>> r = map(lambda x: x*x, itertools.count(1))

结果是什么?

ifilter()

不用多说了,ifilter()就是filter()的惰性实现。
小结

itertools模块提供的全部是处理迭代功能的函数,它们的返回值不是list,而是迭代对象,只有用for循环迭代的时候才真正计算。

Python 相关文章推荐
Python实现动态添加类的属性或成员函数的解决方法
Jul 16 Python
Python中的time模块与datetime模块用法总结
Jun 30 Python
Python中json格式数据的编码与解码方法详解
Jul 01 Python
Python numpy实现数组合并实例(vstack,hstack)
Jan 09 Python
解决nohup执行python程序log文件写入不及时的问题
Jan 14 Python
python实现AES和RSA加解密的方法
Mar 28 Python
彻底理解Python中的yield关键字
Apr 01 Python
Python django框架应用中实现获取访问者ip地址示例
May 17 Python
python实现本地批量ping多个IP的方法示例
Aug 07 Python
Python3视频转字符动画的实例代码
Aug 29 Python
Python脚本去除文件的只读性操作
Mar 05 Python
Python 代码调试技巧示例代码
Aug 11 Python
python使用fileinput模块实现逐行读取文件的方法
Apr 29 #Python
python将字符串转换成数组的方法
Apr 29 #Python
Python中使用hashlib模块处理算法的教程
Apr 28 #Python
简单介绍Python中的struct模块
Apr 28 #Python
在Python中使用base64模块处理字符编码的教程
Apr 28 #Python
使用Python的内建模块collections的教程
Apr 28 #Python
进一步探究Python中的正则表达式
Apr 28 #Python
You might like
PHP垃圾回收机制引用计数器概念分析
2013/06/24 PHP
PHP链接MySQL的常用扩展函数
2014/10/23 PHP
php中使用Ajax时出现Error(c00ce56e)的详细解决方案
2014/11/03 PHP
php 时间time与日期date之间的使用详解及区别
2016/11/07 PHP
javascript引导程序
2008/10/26 Javascript
Prototype 学习 Prototype对象
2009/07/12 Javascript
JavaScript 继承详解 第一篇
2009/08/30 Javascript
鼠标右击事件代码(asp.net后台)
2011/01/27 Javascript
JavaScript 判断日期格式是否正确的实现代码
2011/07/04 Javascript
node.js中的fs.chmod方法使用说明
2014/12/18 Javascript
JS解析XML实例分析
2015/01/30 Javascript
浅谈Javascript数据属性与访问器属性
2016/07/26 Javascript
原生态js,鼠标按下后,经过了那些单元格的简单实例
2016/08/11 Javascript
javascript验证手机号和实现星号(*)代替实例
2016/08/16 Javascript
vue.js国际化 vue-i18n插件的使用详解
2017/07/07 Javascript
zTree获取当前节点的下一级子节点数实例
2017/09/05 Javascript
jquery中done和then的区别(详解)
2017/12/19 jQuery
JavaScript实现Excel表格效果
2020/02/07 Javascript
vue 弹出遮罩层样式实例
2020/07/22 Javascript
jQuery zTree如何改变指定节点文本样式
2020/10/16 jQuery
antd 表格列宽自适应方法以及错误处理操作
2020/10/27 Javascript
[45:52]完美世界DOTA2联赛PWL S3 Forest vs INK ICE 第二场 12.09
2020/12/12 DOTA
初步理解Python进程的信号通讯
2015/04/09 Python
树莓派使用python-librtmp实现rtmp推流h264的方法
2019/07/22 Python
django 多对多表的创建和插入代码实现
2019/09/09 Python
python实现根据给定坐标点生成多边形mask的例子
2020/02/18 Python
Python中logging日志库实例详解
2020/02/19 Python
Python3中小括号()、中括号[]、花括号{}的区别详解
2020/11/15 Python
浅谈Python xlwings 读取Excel文件的正确姿势
2021/02/26 Python
详解CSS3 用border写 空心三角箭头 (两种写法)
2017/09/29 HTML / CSS
CK加拿大官网:Calvin Klein加拿大
2020/03/14 全球购物
面试求职的个人自我评价
2013/11/16 职场文书
化工操作工岗位职责
2014/04/29 职场文书
借条如何写
2015/05/26 职场文书
太空授课观后感
2015/06/17 职场文书
MySql 8.0及对应驱动包匹配的注意点说明
2021/06/23 MySQL