介绍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获取beautifulphoto随机某图片代码实例
Dec 18 Python
浅谈Python黑帽子取代netcat
Feb 10 Python
python opencv 图像尺寸变换方法
Apr 02 Python
Python实现的简单线性回归算法实例分析
Dec 26 Python
python从入门到精通 windows安装python图文教程
May 18 Python
Python生成指定数量的优惠码实操内容
Jun 18 Python
在Python IDLE 下调用anaconda中的库教程
Mar 09 Python
python如何求圆的面积
Jul 01 Python
浅析Python迭代器的高级用法
Jul 16 Python
通过实例解析Python文件操作实现步骤
Sep 21 Python
Python批量获取并保存手机号归属地和运营商的示例
Oct 09 Python
python中str内置函数用法总结
Dec 27 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 number_format() 函数定义和用法
2012/06/01 PHP
关于php程序报date()警告的处理(date_default_timezone_set)
2013/10/22 PHP
PHP实现HTML生成PDF文件的方法
2014/11/07 PHP
PHP中quotemeta()函数的用法讲解
2019/04/04 PHP
JQuery中的$.getJSON 使用说明
2011/03/10 Javascript
5秒后跳转到另一个页面的js代码
2013/10/12 Javascript
js判断设备是否为PC并调整图片大小
2014/02/12 Javascript
一个实用的图片切换支持点击切换和自动轮播
2014/09/09 Javascript
js实现绿白相间竖向网页百叶窗动画切换效果
2015/03/02 Javascript
微信小程序购物商城系统开发系列-目录结构介绍
2016/11/21 Javascript
jQuey将序列化对象在前台显示地实现代码(方法总结)
2016/12/13 Javascript
微信小程序 页面跳转如何实现传值
2017/04/05 Javascript
layui表格实现代码
2017/05/20 Javascript
JS中定位 position 的使用实例代码
2017/08/06 Javascript
vue2.0 实现导航守卫(路由守卫)
2018/05/21 Javascript
学习React中ref的两个demo示例
2018/08/14 Javascript
微信小程序实现多选框全选与取消全选功能示例
2019/05/14 Javascript
Vue中全局变量的定义和使用
2019/06/05 Javascript
如何解决日期函数new Date()浏览器兼容性问题
2019/09/11 Javascript
微信小程序开发(一):服务器获取数据列表渲染操作示例
2020/06/01 Javascript
用Python编写简单的定时器的方法
2015/05/02 Python
python查询sqlite数据表的方法
2015/05/08 Python
Python中的单继承与多继承实例分析
2018/05/10 Python
python自动循环定时开关机(非重启)测试
2019/08/26 Python
Python数据可视化处理库PyEcharts柱状图,饼图,线性图,词云图常用实例详解
2020/02/10 Python
Pycharm连接gitlab实现过程图解
2020/09/01 Python
基于Html5 canvas实现裁剪图片和马赛克功能及又拍云上传图片 功能
2019/07/09 HTML / CSS
饿了么订餐官网:外卖、网上订餐
2019/06/28 全球购物
计算机求职信
2013/12/01 职场文书
光盘行动倡议书
2014/02/02 职场文书
应届毕业生自荐信例文
2014/02/26 职场文书
小学爱国卫生月活动总结
2014/06/30 职场文书
观看焦裕禄观后感
2015/06/09 职场文书
致运动员加油稿
2015/07/21 职场文书
创业计划书之零食店(进口)
2019/09/24 职场文书
matplotlib如何设置坐标轴刻度的个数及标签的方法总结
2021/06/11 Python