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 相关文章推荐
Python3处理文件中每个词的方法
May 22 Python
python 与GO中操作slice,list的方式实例代码
Mar 20 Python
Python subprocess模块详细解读
Jan 29 Python
Python学习之Django的管理界面代码示例
Feb 10 Python
vue.js实现输入框输入值内容实时响应变化示例
Jul 07 Python
Python函数中不定长参数的写法
Feb 13 Python
运用PyTorch动手搭建一个共享单车预测器
Aug 06 Python
Python (Win)readline和tab补全的安装方法
Aug 27 Python
torch 中各种图像格式转换的实现方法
Dec 26 Python
详解selenium + chromedriver 被反爬的解决方法
Oct 28 Python
python 通过pip freeze、dowload打离线包及自动安装的过程详解(适用于保密的离线环境
Dec 14 Python
Python利用机器学习算法实现垃圾邮件的识别
Jun 28 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 URL编码解码函数代码
2009/03/10 PHP
PHP 第二节 数据类型之数值型
2012/04/28 PHP
做了CDN获取用户真实IP的函数代码(PHP与Asp设置方式)
2013/04/13 PHP
ThinkPHP使用心得分享-上传类UploadFile的使用
2014/05/15 PHP
php中preg_match的isU代表什么意思
2015/10/01 PHP
解析 thinkphp 框架中的部分方法
2017/05/07 PHP
laravel框架如何设置公共头和公共尾
2019/10/22 PHP
一个基于jquery的图片切换效果
2010/07/06 Javascript
Asp.net下利用Jquery Ajax实现用户注册检测(验证用户名是否存)
2010/09/12 Javascript
关于JQuery($.load)事件的用法和分析
2013/04/09 Javascript
使用Js让Html中特殊字符不被转义
2013/11/05 Javascript
鼠标移入移出事件改变图片的分辨率的两种方法
2013/12/17 Javascript
JS实现仿Windows7风格的网页右键菜单效果代码
2015/09/11 Javascript
js注入 黑客之路必备!
2016/09/14 Javascript
EasyUI学习之DataGird分页显示数据
2016/12/29 Javascript
微信小程序 video详解及简单实例
2017/01/16 Javascript
基于JQuery及AJAX实现名人名言随机生成器
2017/02/10 Javascript
Vue-resource拦截器判断token失效跳转的实例
2017/10/27 Javascript
JS实现的抛物线运动效果示例
2018/01/30 Javascript
angular6.x中ngTemplateOutlet指令的使用示例
2018/08/09 Javascript
uni-app自定义导航栏按钮|uniapp仿微信顶部导航条功能
2019/11/12 Javascript
[52:03]DOTA2-DPC中国联赛 正赛 Ehome vs iG BO3 第三场 1月31日
2021/03/11 DOTA
Django1.7+python 2.78+pycharm配置mysql数据库教程
2014/11/18 Python
Python提示[Errno 32]Broken pipe导致线程crash错误解决方法
2014/11/19 Python
python3+PyQt5图形项的自定义和交互 python3实现page Designer应用程序
2020/07/20 Python
Python数据相关系数矩阵和热力图轻松实现教程
2020/06/16 Python
Python 如何调试程序崩溃错误
2020/08/03 Python
关于css中margin的值和垂直外边距重叠问题
2020/10/27 HTML / CSS
安踏官方商城:anta.cn
2019/12/16 全球购物
大学学生会竞选演讲稿
2014/04/25 职场文书
团代会宣传工作方案
2014/05/08 职场文书
基层党支部承诺书
2015/04/30 职场文书
2015年保卫科工作总结
2015/05/14 职场文书
公司欠款证明
2015/06/24 职场文书
Python爬虫实战之爬取京东商品数据并实实现数据可视化
2021/06/07 Python
Nginx stream 配置代理(Nginx TCP/UDP 负载均衡)
2021/11/17 Servers