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 文件操作实现代码
Oct 07 Python
python通过floor函数舍弃小数位的方法
Mar 17 Python
python实现多线程抓取知乎用户
Dec 12 Python
Python中关键字nonlocal和global的声明与解析
Mar 12 Python
Python基础教程之内置函数locals()和globals()用法分析
Mar 16 Python
Windows下安装Scrapy
Oct 17 Python
python3+selenium自动化测试框架详解
Mar 17 Python
用python建立两个Y轴的XY曲线图方法
Jul 08 Python
python 列表、字典和集合的添加和删除操作
Dec 16 Python
python爬虫用mongodb的理由
Jul 28 Python
python3代码中实现加法重载的实例
Dec 03 Python
Python用SSH连接到网络设备
Feb 18 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 设计模式之观察者模式介绍
2012/02/22 PHP
处理(php-cgi.exe - FastCGI 进程超过了配置的请求超时时限)的问题
2013/07/03 PHP
php中json_encode UTF-8中文乱码的更好解决方法
2014/09/28 PHP
php版微信公众平台回复中文出现乱码问题的解决方法
2016/09/22 PHP
php is_writable判断文件是否可写实例代码
2016/10/13 PHP
PHP日志LOG类定义与用法示例
2018/09/06 PHP
CI框架网页缓存简单用法分析
2018/12/26 PHP
javascript 定义初始化数组函数
2009/09/07 Javascript
js中同步与异步处理的方法和区别总结
2013/12/25 Javascript
微信小程序 textarea 详解及简单使用方法
2016/12/05 Javascript
js return返回多个值,通过对象的属性访问方法
2017/02/21 Javascript
微信小程序页面传值实例分析
2017/04/19 Javascript
基于jQuery封装的分页组件
2017/06/26 jQuery
详解Vue路由钩子及应用场景(小结)
2017/11/07 Javascript
vue如何通过id从列表页跳转到对应的详情页
2018/05/01 Javascript
微信小程序网络请求封装示例
2018/07/24 Javascript
vue-router的使用方法及含参数的配置方法
2018/11/13 Javascript
实例讲解JS中pop使用方法
2019/01/27 Javascript
JavaScript 中的执行上下文和执行栈实例讲解
2021/02/25 Javascript
python多线程编程中的join函数使用心得
2014/09/02 Python
Python编码爬坑指南(必看)
2016/06/10 Python
使用Python进行二进制文件读写的简单方法(推荐)
2016/09/12 Python
python写入并获取剪切板内容的实例
2018/05/31 Python
Python中的十大图像处理工具(小结)
2019/06/10 Python
python global关键字的用法详解
2019/09/05 Python
Python OpenCV视频截取并保存实现代码
2019/11/30 Python
Pytorch提取模型特征向量保存至csv的例子
2020/01/03 Python
python 利用matplotlib在3D空间绘制二次抛物面的案例
2021/02/06 Python
浅谈基于Canvas的手绘风格图形库Rough.js
2018/03/19 HTML / CSS
EMU Australia澳大利亚官网:澳大利亚本土雪地靴品牌
2019/07/24 全球购物
高中化学教学反思
2014/01/13 职场文书
餐厅周年庆活动方案
2014/08/25 职场文书
党的群众路线教育实践活动实施方案
2014/10/31 职场文书
php微信小程序解包过程实例详解
2021/03/31 PHP
SQL基础的查询语句
2021/11/11 MySQL
Python可视化学习之matplotlib内置单颜色
2022/02/24 Python