介绍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爬虫之使用Scrapy框架编写爬虫
Nov 07 Python
详解python单例模式与metaclass
Jan 15 Python
Python ldap实现登录实例代码
Sep 30 Python
Python Xml文件添加字节属性的方法
Mar 31 Python
python批量导入数据进Elasticsearch的实例
May 30 Python
Selenium鼠标与键盘事件常用操作方法示例
Aug 13 Python
pygame游戏之旅 调用按钮实现游戏开始功能
Nov 21 Python
Python企业编码生成系统之系统主要函数设计详解
Jul 26 Python
Python time库基本使用方法分析
Dec 13 Python
基于Python pyecharts实现多种图例代码解析
Aug 10 Python
基于python实现操作redis及消息队列
Aug 27 Python
如何基于Python实现word文档重新排版
Sep 29 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 正则表达式常用函数
2014/08/17 PHP
PHP+jQuery 注册模块开发详解
2014/10/14 PHP
微信公众平台开发关注及取消关注事件的方法
2014/12/23 PHP
Smarty高级应用之缓存操作技巧分析
2016/05/14 PHP
Yii多表联合查询操作详解
2016/06/02 PHP
Windows上php5.6操作mongodb数据库示例【配置、连接、获取实例】
2019/02/13 PHP
PHP将整数数字转换为罗马数字实例分享
2019/03/17 PHP
PHP使用POP3读取邮箱接收邮件的示例代码
2020/07/08 PHP
Google Map Api和GOOGLE Search Api整合实现代码
2009/07/18 Javascript
基于jquery插件实现常见的幻灯片效果
2013/11/01 Javascript
html5+javascript制作简易画板附图
2014/04/25 Javascript
GitHub上一些实用的JavaScript的文件压缩解压缩库推荐
2016/03/13 Javascript
js 将图片连接转换成base64格式的简单实例
2016/08/10 Javascript
jQuery中的select操作详解
2016/11/29 Javascript
JS实现的简单四则运算计算器功能示例
2017/09/27 Javascript
Angular搜索 过滤 批量删除 添加 表单验证功能集锦(实例代码)
2017/10/25 Javascript
Vue中使用create-keyframe-animation与动画钩子完成复杂动画
2019/04/09 Javascript
详解vue中使用axios对同一个接口连续请求导致返回数据混乱的问题
2019/11/06 Javascript
vue keep-alive的简单总结
2021/01/25 Vue.js
Python datetime时间格式化去掉前导0
2014/07/31 Python
Python Nose框架编写测试用例方法
2017/10/26 Python
pandas按若干个列的组合条件筛选数据的方法
2018/04/11 Python
详解Python中的四种队列
2018/05/21 Python
Python数据预处理之数据规范化(归一化)示例
2019/01/08 Python
python实现串口自动触发工作的示例
2019/07/02 Python
在pycharm下设置自己的个性模版方法
2019/07/15 Python
python实现滑雪者小游戏
2020/02/22 Python
Django跨域资源共享问题(推荐)
2020/03/09 Python
Python实现定时监测网站运行状态的示例代码
2020/09/30 Python
雅诗兰黛香港官网:Estee Lauder香港
2017/09/26 全球购物
总经理职责范文
2013/11/08 职场文书
房屋出租协议书
2014/04/10 职场文书
学校四风对照检查材料
2014/08/28 职场文书
道歉的话怎么说
2015/05/12 职场文书
结婚幸福感言
2015/08/01 职场文书
Mac电脑OS系统下安装Nginx的详细教程
2022/04/14 Servers