介绍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 相关文章推荐
python3.3实现乘法表示例
Feb 07 Python
python如何使用正则表达式的前向、后向搜索及前向搜索否定模式详解
Nov 08 Python
Python之多线程爬虫抓取网页图片的示例代码
Jan 10 Python
对Python 语音识别框架详解
Dec 24 Python
使用GitHub和Python实现持续部署的方法
May 09 Python
Django 请求Request的具体使用方法
Nov 11 Python
Django Admin设置应用程序及模型顺序方法详解
Apr 01 Python
基于python实现模拟数据结构模型
Jun 12 Python
python 提高开发效率的5个小技巧
Oct 19 Python
Python join()函数原理及使用方法
Nov 14 Python
Python Pandas知识点之缺失值处理详解
May 11 Python
教你用Python爬取英雄联盟皮肤原画
Jun 13 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 curl模拟post提交数据示例
2013/12/31 PHP
ThinkPHP自动完成中使用函数与回调方法实例
2014/11/29 PHP
微信红包随机生成算法php版
2016/07/21 PHP
PHP设计模式之原型设计模式原理与用法分析
2018/04/25 PHP
php判断/计算闰年的方法小结【三种方法】
2019/07/06 PHP
Yii框架核心组件类实例详解
2019/08/06 PHP
laravel 验证错误信息到 blade模板的方法
2019/09/29 PHP
JavaScript中的变量声明早于赋值分析
2012/03/01 Javascript
js不完美解决click和dblclick事件冲突问题
2012/07/16 Javascript
javascript面向对象快速入门实例
2015/01/13 Javascript
简介AngularJS中$http服务的用法
2016/02/06 Javascript
JavaScript动态生成二维码图片
2016/04/20 Javascript
jQuery Ajax传值到Servlet出现乱码问题的解决方法
2016/10/09 Javascript
KnockoutJS 3.X API 第四章之表单submit、enable、disable绑定
2016/10/10 Javascript
Mac 安装 nodejs方法(图文详细步骤)
2017/10/30 NodeJs
vue-cli中的babel配置文件.babelrc实例详解
2018/02/22 Javascript
浅谈Vue响应式(数组变异方法)
2018/05/07 Javascript
Vue 路由 过渡动效 数据获取方法
2018/07/31 Javascript
js中的数组对象排序分析
2018/12/11 Javascript
微信小程序下拉框搜索功能的实现方法
2019/07/31 Javascript
layui动态表头的实现代码
2019/08/22 Javascript
微信小程序实现音频文件播放进度的实例代码
2020/03/02 Javascript
Python3内置模块之json编解码方法小结【推荐】
2020/12/09 Python
django框架单表操作之增删改实例分析
2019/12/16 Python
Python3列表List入门知识附实例
2020/02/09 Python
新手入门学习python Numpy基础操作
2020/03/02 Python
python 实现客户端与服务端的通信
2020/12/23 Python
苹果中国官方网站:Apple中国
2016/07/22 全球购物
俄罗斯最大的灯具网站:Fandeco
2020/03/14 全球购物
会计师事务所审计实习自我鉴定
2013/09/20 职场文书
2014新课程改革心得体会
2014/03/10 职场文书
一份恶作剧的检讨书
2014/09/13 职场文书
2015年党员创先争优承诺书
2015/01/22 职场文书
2016年小学生新年寄语
2015/08/18 职场文书
python opencv检测直线 cv2.HoughLinesP的实现
2021/06/18 Python
试了下Golang实现try catch的方法
2021/07/01 Golang