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编写简单的HTML页面合并脚本
Jul 11 Python
python监控键盘输入实例代码
Feb 09 Python
Python2中文处理纪要的实现方法
Mar 10 Python
基于Python3.6+splinter实现自动抢火车票
Sep 25 Python
python取数作为临时极大值(极小值)的方法
Oct 15 Python
对python 调用类属性的方法详解
Jul 02 Python
Pandas时间序列:重采样及频率转换方式
Dec 26 Python
django rest framework serializer返回时间自动格式化方法
Mar 31 Python
通过案例解析python鸭子类型相关原理
Oct 10 Python
python利用opencv保存、播放视频
Nov 02 Python
pytorch 实现多个Dataloader同时训练
May 29 Python
python利用while求100内的整数和方式
Nov 07 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
收听困难?教您超简便短波广播抗干扰方法!
2021/03/01 无线电
PHP中::、-&amp;gt;、self、$this几种操作符的区别介绍
2013/04/24 PHP
thinkphp5.0整合phpsocketio完整攻略(绕坑)
2018/10/12 PHP
javascript删除数组元素并且数组长度减小的简单实例
2014/02/14 Javascript
JsRender实用入门教程
2014/10/31 Javascript
jquery Validation表单验证使用详解
2020/09/12 Javascript
jQuery实现html元素拖拽
2015/07/21 Javascript
详解Bootstrap创建表单的三种格式(一)
2016/01/04 Javascript
不能不知道的10个angularjs英文学习网站
2016/03/23 Javascript
js编写当天简单日历效果【实现代码】
2016/05/03 Javascript
详解Vue路由History mode模式中页面无法渲染的原因及解决
2017/09/28 Javascript
jQuery选择器之子元素过滤选择器
2017/09/28 jQuery
JS实现移动端触屏拖拽功能
2018/07/31 Javascript
jQuery-ui插件sortable实现自由拖动排序
2018/12/01 jQuery
Angular封装搜索框组件操作示例
2019/04/25 Javascript
jquery 遍历hash操作示例【基于ajax交互】
2019/10/12 jQuery
详解js location.href和window.open的几种用法和区别
2019/12/02 Javascript
JS window对象简单操作完整示例
2020/01/14 Javascript
vue v-for出来的列表,点击某个li使得当前被点击的li字体变红操作
2020/07/17 Javascript
React实现todolist功能
2020/12/28 Javascript
[31:29]完美世界DOTA2联赛PWL S3 INK ICE vs Magma 第一场 12.20
2020/12/23 DOTA
Python判断文件和字符串编码类型的实例
2017/12/21 Python
对Python中创建进程的两种方式以及进程池详解
2019/01/14 Python
Python删除n行后的其他行方法
2019/01/28 Python
Python OOP类中的几种函数或方法总结
2019/02/22 Python
python 求一个列表中所有元素的乘积实例
2019/06/11 Python
分析运行中的 Python 进程详细解析
2019/06/22 Python
python爬取Ajax动态加载网页过程解析
2019/09/05 Python
如何使用html5与css3完成google涂鸦动画
2012/12/16 HTML / CSS
Darphin迪梵官网: 来自巴黎,植物和精油调制的护肤品牌
2016/10/11 全球购物
猫咪家具:CatsPlay
2018/11/03 全球购物
学校运动会开幕演讲稿
2014/01/04 职场文书
小学生安全保证书
2014/02/01 职场文书
学校教学工作总结2015
2015/05/19 职场文书
英语投诉信范文
2015/07/03 职场文书
2015年秋季运动会前导词
2015/07/20 职场文书