Python标准库之循环器(itertools)介绍


Posted in Python onNovember 25, 2014

在循环对象和函数对象中,我们了解了循环器(iterator)的功能。循环器是对象的容器,包含有多个对象。通过调用循环器的next()方法 (__next__()方法,在Python 3.x中),循环器将依次返回一个对象。直到所有的对象遍历穷尽,循环器将举出StopIteration错误。

在for i in iterator结构中,循环器每次返回的对象将赋予给i,直到循环结束。使用iter()内置函数,我们可以将诸如表、字典等容器变为循环器。比如:

for i in iter([2, 4, 5, 6]):

    print(i)

标准库中的itertools包提供了更加灵活的生成循环器的工具。这些工具的输入大都是已有的循环器。另一方面,这些工具完全可以自行使用Python实现,该包只是提供了一种比较标准、高效的实现方式。这也符合Python“只有且最好只有解决方案”的理念。

# import the tools

from itertools import *

无穷循环器

count(5, 2)     #从5开始的整数循环器,每次增加2,即5, 7, 9, 11, 13, 15 ...
cycle('abc')    #重复序列的元素,既a, b, c, a, b, c ...
repeat(1.2)     #重复1.2,构成无穷循环器,即1.2, 1.2, 1.2, ...

repeat也可以有一个次数限制:

repeat(10, 5)   #重复10,共重复5次

函数式工具

函数式编程是将函数本身作为处理对象的编程范式。在Python中,函数也是对象,因此可以轻松的进行一些函数式的处理,比如map(), filter(), reduce()函数。

itertools包含类似的工具。这些函数接收函数作为参数,并将结果返回为一个循环器。

比如:

from itertools import *

rlt = imap(pow, [1, 2, 3], [1, 2, 3])

for num in rlt:

    print(num)

上面显示了imap函数。该函数与map()函数功能相似,只不过返回的不是序列,而是一个循环器。包含元素1, 4, 27,即1**1, 2**2, 3**3的结果。函数pow(内置的乘方函数)作为第一个参数。pow()依次作用于后面两个列表的每个元素,并收集函数结果,组成返回的循环器。

此外,还可以用下面的函数:

starmap(pow, [(1, 1), (2, 2), (3, 3)])

pow将依次作用于表的每个tuple。

ifilter函数与filter()函数类似,只是返回的是一个循环器。

ifilter(lambda x: x > 5, [2, 3, 5, 6, 7]

将lambda函数依次作用于每个元素,如果函数返回True,则收集原来的元素。6, 7

此外,

ifilterfalse(lambda x: x > 5, [2, 3, 5, 6, 7])

与上面类似,但收集返回False的元素。2, 3, 5
takewhile(lambda x: x < 5, [1, 3, 6, 7, 1])

当函数返回True时,收集元素到循环器。一旦函数返回False,则停止。1, 3
dropwhile(lambda x: x < 5, [1, 3, 6, 7, 1])

当函数返回False时,跳过元素。一旦函数返回True,则开始收集剩下的所有元素到循环器。6, 7, 1。

组合工具

我们可以通过组合原有循环器,来获得新的循环器。

chain([1, 2, 3], [4, 5, 7])      # 连接两个循环器成为一个。1, 2, 3, 4, 5, 7

product('abc', [1, 2])   # 多个循环器集合的笛卡尔积。相当于嵌套循环  

for m, n in product('abc', [1, 2]):

    print m, n

permutations('abc', 2)   # 从'abcd'中挑选两个元素,比如ab, bc, ... 将所有结果排序,返回为新的循环器。

注意,上面的组合分顺序,即ab, ba都返回。

combinations('abc', 2)   # 从'abcd'中挑选两个元素,比如ab, bc, ... 将所有结果排序,返回为新的循环器。

注意,上面的组合不分顺序,即ab, ba的话,只返回一个ab。

combinations_with_replacement('abc', 2) # 与上面类似,但允许两次选出的元素重复。即多了aa, bb, cc。

groupby()

将key函数作用于原循环器的各个元素。根据key函数结果,将拥有相同函数结果的元素分到一个新的循环器。每个新的循环器以函数返回结果为标签。

这就好像一群人的身高作为循环器。我们可以使用这样一个key函数: 如果身高大于180,返回"tall";如果身高底于160,返回"short";中间的返回"middle"。最终,所有身高将分为三个循环器,即"tall", "short", "middle"。

def height_class(h):

     if h > 180:

        return "tall"

     elif h < 160:

        return "short"

     else:

        return "middle"friends = [191, 158, 159, 165, 170, 177, 181, 182, 190] friends = sorted(friends, key = height_class)

 

for m, n in groupby(friends, key = height_class):

    print(m)

    print(list(n))

注意,groupby的功能类似于UNIX中的uniq命令。分组之前需要使用sorted()对原循环器的元素,根据key函数进行排序,让同组元素先在位置上靠拢。

其它工具

compress('ABCD', [1, 1, 1, 0])  # 根据[1, 1, 1, 0]的真假值情况,选择第一个参数'ABCD'中的元素。A, B, C
islice()                        # 类似于slice()函数,只是返回的是一个循环器
izip()                          # 类似于zip()函数,只是返回的是一个循环器。

总结

itertools的工具都可以自行实现。itertools只是提供了更加成形的解决方案。

Python 相关文章推荐
浅谈Python程序与C++程序的联合使用
Apr 07 Python
介绍Python的Django框架中的QuerySets
Apr 20 Python
Selenium定时刷新网页的实现代码
Oct 31 Python
python-itchat 统计微信群、好友数量,及原始消息数据的实例
Feb 21 Python
详解利用Python scipy.signal.filtfilt() 实现信号滤波
Jun 05 Python
python yield关键词案例测试
Oct 15 Python
Python读取YAML文件过程详解
Dec 30 Python
Tensorflow中tf.ConfigProto()的用法详解
Feb 06 Python
python+opencv实现移动侦测(帧差法)
Mar 20 Python
基于selenium及python实现下拉选项定位select
Jul 22 Python
python的dict判断key是否存在的方法
Dec 09 Python
PyTorch中的torch.cat简单介绍
Mar 17 Python
Python标准库之sqlite3使用实例
Nov 25 #Python
Python标准库内置函数complex介绍
Nov 25 #Python
Python XML RPC服务器端和客户端实例
Nov 22 #Python
Python实现读取目录所有文件的文件名并保存到txt文件代码
Nov 22 #Python
python进程类subprocess的一些操作方法例子
Nov 22 #Python
Python读取环境变量的方法和自定义类分享
Nov 22 #Python
Python中的引用和拷贝浅析
Nov 22 #Python
You might like
推荐php模板技术[转]
2007/01/04 PHP
PHP 开发环境配置(测试开发环境)
2010/04/28 PHP
PHP 时间日期操作实战
2011/08/26 PHP
php实现上传图片生成缩略图示例
2014/04/13 PHP
ThinkPHP表单自动验证实例
2014/10/13 PHP
php中执行系统命令的方法
2015/03/21 PHP
Mac下php 5升级到php 7的步骤详解
2017/04/26 PHP
Yii2结合Workerman的websocket示例详解
2018/09/10 PHP
PHP面向对象类型约束用法分析
2019/06/12 PHP
解密效果
2006/06/23 Javascript
juqery 学习之四 筛选过滤
2010/11/30 Javascript
关闭ie窗口清除Session的解决方法
2014/01/10 Javascript
详解Javascript动态操作CSS
2014/12/08 Javascript
浅析jquery unbind()方法移除元素绑定的事件
2016/05/24 Javascript
js改变html的原有内容实现方法
2016/10/05 Javascript
使用jQuery实现鼠标点击左右按钮滑动切换
2017/08/04 jQuery
简述ES6新增关键字let与var的区别
2019/08/23 Javascript
JS代码检查工具ESLint介绍与使用方法
2020/02/04 Javascript
vue v-for 点击当前行,获取当前行数据及event当前事件对象的操作
2020/09/10 Javascript
jQuery插件实现图片轮播效果
2020/10/19 jQuery
javascript实现拼图游戏
2021/01/29 Javascript
python使用pil生成图片验证码的方法
2015/05/08 Python
Python爬取三国演义的实现方法
2016/09/12 Python
对python的输出和输出格式详解
2018/12/08 Python
在pycharm中使用git版本管理以及同步github的方法
2019/01/16 Python
Python数据报表之Excel操作模块用法分析
2019/03/11 Python
对Pytorch中Tensor的各种池化操作解析
2020/01/03 Python
Python 如何对文件目录操作
2020/07/10 Python
日本索尼音乐商店:Sony Music Shop
2018/07/17 全球购物
护理专业毕业生自我鉴定
2013/10/08 职场文书
模具设计与制造专业应届生求职信
2013/10/18 职场文书
中国梦演讲稿5分钟
2014/08/19 职场文书
以权谋私检举信范文
2015/03/02 职场文书
死亡诗社观后感
2015/06/05 职场文书
2016党员党章学习心得体会
2016/01/14 职场文书
nginx location中多个if里面proxy_pass的方法
2021/03/31 Servers