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实现保证只能运行一个脚本实例
Jun 24 Python
利用TensorFlow训练简单的二分类神经网络模型的方法
Mar 05 Python
Python对多属性的重复数据去重实例
Apr 18 Python
解决python 输出是省略号的问题
Apr 19 Python
对python append 与浅拷贝的实例讲解
May 04 Python
Python实现多条件筛选目标数据功能【测试可用】
Jun 13 Python
Python+selenium 获取浏览器窗口坐标、句柄的方法
Oct 14 Python
Python在cmd上打印彩色文字实现过程详解
Aug 07 Python
基于Python+Appium实现京东双十一自动领金币功能
Oct 31 Python
python实现b站直播自动发送弹幕功能
Feb 20 Python
Pytest allure 命令行参数的使用
Apr 18 Python
Django一小时写出账号密码管理系统
Apr 29 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
ThinkPHP之用户注册登录留言完整实例
2014/07/22 PHP
php过滤HTML标签、属性等正则表达式汇总
2014/09/22 PHP
php使用timthumb生成缩略图的方法
2016/01/22 PHP
PHP封装的page分页类定义与用法完整示例
2018/12/24 PHP
图片格式的JavaScript和CSS速查手册
2007/08/20 Javascript
js 方法实现返回多个数据的代码
2009/04/30 Javascript
jQuery 标题的自动翻转实现代码
2009/10/14 Javascript
JS 去除Array中的null值示例代码
2013/11/20 Javascript
Jquery对数组的操作技巧整理
2014/03/25 Javascript
jquery禁用右键示例
2014/04/28 Javascript
js实现获取焦点后光标在字符串后
2014/09/17 Javascript
jQuery简单实现提交数据出现loading进度条的方法
2016/03/29 Javascript
使用bootstrap3开发响应式网站
2016/05/12 Javascript
javascript中mouseenter与mouseover的异同
2017/06/06 Javascript
详解如何去除vue项目中的#——History模式
2017/10/13 Javascript
jQuery替换节点元素的操作方法
2018/03/18 jQuery
Vue.js+cube-ui(Scroll组件)实现类似头条效果的横向滚动导航条
2019/06/24 Javascript
Python代码调试的几种方法总结
2015/04/15 Python
Python的Django框架中从url中捕捉文本的方法
2015/07/20 Python
python对文件目录的操作方法实例总结
2019/06/24 Python
python 判断字符串中是否含有汉字或非汉字的实例
2019/07/15 Python
Django模板Templates使用方法详解
2019/07/19 Python
Python爬虫运用正则表达式的方法和优缺点
2019/08/25 Python
python实现将视频按帧读取到自定义目录
2019/12/10 Python
Python使用pyyaml模块处理yaml数据
2020/04/14 Python
python3.6.8 + pycharm + PyQt5 环境搭建的图文教程
2020/06/11 Python
DRF使用simple JWT身份验证的实现
2021/01/14 Python
借助HTML5 Canvas API制作一个简单的猜字游戏
2016/03/25 HTML / CSS
高中生自我鉴定范文
2013/10/30 职场文书
执行总经理岗位职责
2014/02/03 职场文书
《要下雨了》教学反思
2014/02/17 职场文书
潘婷洗发水广告词
2014/03/14 职场文书
个人银行贷款担保书
2014/04/01 职场文书
保洁公司服务承诺书
2014/05/28 职场文书
2015幼儿园新学期寄语
2015/02/27 职场文书
简历自荐信范文
2015/03/09 职场文书