Python中itertools的用法详解


Posted in Python onFebruary 07, 2020

iterator

循环器(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实现,该包只是提供了一种比较标准、高效的实现方式。

# import the tools
from itertools import *

无穷循环器

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

函数式工具

函数式编程是将函数本身作为处理对象的编程范式。在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。

组合工具

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

循环器和笛卡尔乘积

# 连接两个循环器成为一个。1, 2, 3, 4, 5, 7
chain([1, 2, 3], [4, 5, 7]) 
# 多个循环器集合的笛卡尔积,相当于嵌套循环。  
product('abc', [1, 2])

样例如下

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

'''
a 1
a 2
b 1
b 2
c 1
c 2
'''

排列与组合

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

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

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

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

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

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函数进行排序,让同组元素先在位置上靠拢。

其它工具

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

总结

以上所述是小编给大家介绍的Python中itertools的用法详解,希望对大家有所帮助!

Python 相关文章推荐
Python中用于去除空格的三个函数的使用小结
Apr 07 Python
Android分包MultiDex策略详解
Oct 30 Python
分享6个隐藏的python功能
Dec 07 Python
TensorFLow用Saver保存和恢复变量
Mar 10 Python
python 爬虫一键爬取 淘宝天猫宝贝页面主图颜色图和详情图的教程
May 22 Python
Python爬虫框架Scrapy基本用法入门教程
Jul 26 Python
pandas把所有大于0的数设置为1的方法
Jan 26 Python
Flask教程之重定向与错误处理实例分析
Aug 01 Python
python django生成迁移文件的实例
Aug 31 Python
python打印直角三角形与等腰三角形实例代码
Oct 20 Python
tensorflow安装成功import tensorflow 出现问题
Apr 16 Python
Python制作简单的剪刀石头布游戏
Dec 10 Python
Python转换itertools.chain对象为数组的方法
Feb 07 #Python
已安装tensorflow-gpu,但keras无法使用GPU加速的解决
Feb 07 #Python
python十进制转二进制的详解
Feb 07 #Python
基于Tensorflow使用CPU而不用GPU问题的解决
Feb 07 #Python
python实现ip地址的包含关系判断
Feb 07 #Python
tensorflow:指定gpu 限制使用量百分比,设置最小使用量的实现
Feb 06 #Python
Tensorflow中tf.ConfigProto()的用法详解
Feb 06 #Python
You might like
PHP过滤★等特殊符号的正则
2014/01/27 PHP
使用PHP接受文件并获得其后缀名的方法
2015/08/05 PHP
启用OPCache提高PHP程序性能的方法
2019/03/21 PHP
对google个性主页的拖拽效果的js的完整注释[转]
2007/04/10 Javascript
如何使用Javascript正则表达式来格式化XML内容
2013/07/04 Javascript
js实现分享到随页面滚动而滑动效果的方法
2015/04/10 Javascript
四种参数传递的形式——URL,超链接,js,form表单
2015/07/24 Javascript
基于JS代码实现当鼠标悬停表格上显示这一格的全部内容
2016/06/12 Javascript
原生javascript实现的ajax异步封装功能示例
2016/11/03 Javascript
JS前向后瞻正则表达式定义与用法示例
2016/12/27 Javascript
EasyUI Datebox 日期验证之开始日期小于结束时间
2017/05/19 Javascript
Three.js利用Detector.js插件如何实现兼容性检测详解
2017/09/26 Javascript
vue-lazyload图片延迟加载插件的实例讲解
2018/02/09 Javascript
vue最简单的前后端交互示例详解
2018/10/11 Javascript
浅谈HTTP 缓存的那些事儿
2018/10/17 Javascript
Vue + Elementui实现多标签页共存的方法
2019/06/12 Javascript
Ajax请求时无法重定向的问题解决代码详解
2019/06/21 Javascript
js实现的格式化数字和金额功能简单示例
2019/07/30 Javascript
vue图片上传组件使用详解
2019/12/23 Javascript
vue实现登录拦截
2020/06/29 Javascript
小结Python用fork来创建子进程注意事项
2014/07/03 Python
python游戏开发之视频转彩色字符动画
2019/04/26 Python
Python直接赋值、浅拷贝与深度拷贝实例分析
2019/06/18 Python
使用python创建生成动态链接库dll的方法
2020/05/09 Python
canvas画布实现手写签名效果的示例代码
2019/04/23 HTML / CSS
阿迪达斯加拿大官网:Adidas加拿大
2016/08/25 全球购物
英国旅游额外服务市场领导者:Holiday Extras(机场停车场、酒店、接送等)
2017/10/07 全球购物
意大利时尚奢侈品店:D’Aniello Boutique
2021/01/19 全球购物
速卖通欧盟:Aliexpress EU
2020/08/19 全球购物
sleep()方法和wait()方法的区别是什么
2012/11/17 面试题
小学家长评语大全
2014/04/16 职场文书
论文诚信承诺书
2014/05/23 职场文书
教师批评与自我批评(群众路线)
2014/10/15 职场文书
应急管理工作总结2015
2015/05/04 职场文书
集结号观后感
2015/06/08 职场文书
MySQL transaction事务安全示例讲解
2022/06/21 MySQL