Python函数式编程中itertools模块详解


Posted in Python onSeptember 15, 2021

容器与可迭代对象

在正式开始前先补充一些基本概念在 Python 中存在容器 与 可迭代对象

  • 容器:用来存储多个元素的数据结构,例如 列表,元组,字典,集合等内容;
  • 可迭代对象:实现了 __iter__ 方法的对象就叫做可迭代对象。

从可迭代对象中还衍生出 迭代器 与 生成器:

  • 迭代器:既实现了 __iter__,也实现了 __next__ 方法的对象叫做迭代器;
  • 生成器:具有 yield 关键字的函数都是生成器。

这样就比较清楚了,可迭代对象的范围要大于容器。而且可迭代对象只能使用一次,使用完毕再获取值就会提示 StopIteration 异常。

除此之外,可迭代对象还有一些限制:

  • 不能对可迭代对象使用 len 函数;
  • 可以使用 next 方法处理可迭代对象,容器也可以通过 iter 函数转换成迭代器;
  • for 语句会自动调用容器的 iter 函数,所以容器也能被循环迭代。

count() 函数

count 函数一般与 range 函数对比学习,例如 range 函数需要定义生成范围的下限,上限与步长可选,而 count 函数不同,指定下限与步长,上限值不用声明。

函数原型声明如下

count(start=0, step=1) --> count object

测试代码如下,其中必须添加跳出循环的判定条件,否则代码会一直运行下去。

from itertools import count
a = count(5, 10)
for i in a:
    print(i)
    if i > 100:
        break

除此之外,count 函数还接收非整数参数,所以下述代码中定义的也是正确的。

from itertools import count
a = count(0.5, 0.1)
for i in a:
    print(i)
    if i > 100:
        break

cycle 函数

用 cycle 函数可以循环一组值,测试代码如下所示:

from itertools import cycle
x = cycle('梦想橡皮擦abcdf')
for i in range(5):
    print(next(x), end=" ")
print("\n")
print("*" * 100)
for i in range(5):
    print(next(x), end=" ")

代码输出如下内容:

梦 想 橡 皮 擦

****************************************************************************************************
a b c d f

可以看到 cycle 函数与 for 循环非常类似。

repeat 函数

repeat 函数用于重复返回某个值,官方给出的函数描述如下所示:

class repeat(object):
    """
    repeat(object [,times]) -> create an iterator which returns the object
    for the specified number of times.  If not specified, returns the object
    endlessly.

进行一下简单的测试,看一下效果:

from itertools import repeat
x = repeat('橡皮擦')
for i in range(5):
    print(next(x), end=" ")
print("\n")
print("*" * 100)
for i in range(5):
    print(next(x), end=" ")

怎么看这个函数,都好像没有太大用处。

enumerate 函数,添加序号

这个函数在前面的文章中,已经进行过简单介绍,并且该函数在 __builtins__ 包中,所以不再过多说明,基本格式如下所示:

enumerate(sequence, [start=0])

其中 start 参数是下标起始位置。

accumulate 函数

该函数基于给定的函数返回一个可迭代对象,默认是累加效果,即第二个参数为 operator.add,测试代码如下:

from itertools import accumulate
data = [1, 2, 3, 4, 5]
# 计算累积和
print(list(accumulate(data)))  # [1, 3, 6, 10, 15]

针对上述代码,修改为累积。

from itertools import accumulate
import operator
data = [1, 2, 3, 4, 5]
# 计算累积
print(list(accumulate(data, operator.mul)))

除此之外,第二个参数还可以为 max,min 等函数,例如下述代码:

from itertools import accumulate
data = [1, 4, 3, 2, 5]
print(list(accumulate(data, max)))

代码输出如下内容,其实是将 data 里面的任意两个值进行了比较,然后留下最大的值。

[1, 4, 4, 4, 5]

chain 与 groupby 函数

chain 函数用于将多个迭代器组合为单个迭代器,而 groupby 可以将一个迭代器且分为多个子迭代器。

首先展示一下 groupby 函数的应用:

from itertools import groupby
a = list(groupby('橡橡皮皮擦擦'))
print(a)

输出内容如下所示:

[('橡', <itertools._grouper object at 0x0000000001DD9438>),
('皮', <itertools._grouper object at 0x0000000001DD9278>),
('擦', <itertools._grouper object at 0x00000000021FF710>)]

为了使用 groupby 函数,建议先对原列表进行排序,因为它是有点像切片一样,发现不同的就分出一个迭代器。

chain 函数的用法如下,将多个迭代对象进行拼接:

from itertools import groupby, chain
a = list(chain('ABC', 'AAA', range(1,3)))
print(a)

zip_longest 与 zip

zip 函数在之前的博客中已经进行过说明,zip_longest 与 zip 的区别就是,zip 返回的结果以最短的序列为准,而 zip_longest 以最长的为准。

测试代码如下,自行比对结果即可。

from itertools import zip_longest
a = list(zip('ABC', range(5), [10, 20, 30, 40]))
print(a)
a = list(zip_longest('ABC', range(5), [10, 20, 30, 40]))
print(a)

zip_logest 如果碰到长度不一致的序列,缺少部分会填充 None。

tee 函数

tee 函数可以克隆可迭代对象,产出多个生成器,每个生成器都可以产出输入的各个元素。

from itertools import tee
a = list(tee('橡皮擦'))
print(a)

compress 函数

该函数通过**谓词(是否,True/False)**来确定对某个元素的取舍问题,最简单的代码如下所示:

from itertools import compress
a = list(compress('橡皮擦', (0, 1, 1)))
print(a)

islice、dropwhile、takewhile、filterfalse、filter

这几个函数都是从输入的可迭代对象中获取一个子集,而且不修改元素本身。

本部分只罗列各个函数的原型声明,具体用法直接参考使用即可。

islice(iterable, stop) --> islice object
islice(iterable, start, stop[, step]) --> islice object
dropwhile(predicate, iterable) --> dropwhile object
takewhile(predicate, iterable) --> takewhile object
filterfalse(function or None, sequence) --> filterfalse object

其中只有 filterfalse 中的参数是函数在前,序列在后。

测试代码如下,尤其注意第一个参数是 callable 即函数。

from itertools import islice, dropwhile, takewhile, filterfalse
a = list(filterfalse(lambda x: x in ["皮", "擦"], '橡皮擦'))
print(a)

总结

以上内容就是本文的全部内容,在使用无限迭代器函数 count,cycle,repeat 的时候,一定要注意即使停止。

本篇文章就到这里了,希望能够给你带来帮助,也希望您能够多多关注三水点靠木的更多内容!

Python 相关文章推荐
python追加元素到列表的方法
Jul 28 Python
R语言 vs Python对比:数据分析哪家强?
Nov 17 Python
Python 读取图片文件为矩阵和保存矩阵为图片的方法
Apr 27 Python
详解利用django中间件django.middleware.csrf.CsrfViewMiddleware防止csrf攻击
Oct 09 Python
python3实现zabbix告警推送钉钉的示例
Feb 20 Python
TensorFlow车牌识别完整版代码(含车牌数据集)
Aug 05 Python
OpenCV灰度化之后图片为绿色的解决
Dec 01 Python
详解Django中的FBV和CBV对比分析
Mar 01 Python
pandas 实现将NaN转换为None
May 14 Python
Python3 类型标注支持操作
Jun 02 Python
教你怎么用Python selenium操作浏览器对象的基础API
Jun 23 Python
python数字转对应中文的方法总结
Aug 02 Python
Python编程中Python与GIL互斥锁关系作用分析
Sep 15 #Python
Python3.10的一些新特性原理分析
Sep 15 #Python
一篇文章带你了解Python和Java的正则表达式对比
Sep 15 #Python
Python编程编写完善的命令行工具
Sep 15 #Python
python可视化之颜色映射详解
python的变量和简单数字类型详解
Sep 15 #Python
深入浅析Django MTV模式
You might like
PHP 源代码压缩小工具
2009/12/22 PHP
PHP5中使用PDO连接数据库的方法
2010/08/01 PHP
深入PHP magic quotes的详解
2013/06/17 PHP
PHP中trait使用方法详细介绍
2017/05/21 PHP
php实现微信分享朋友链接功能
2019/02/18 PHP
用JS判断IE版本的代码 超管用!
2011/08/09 Javascript
javascript真的不难-回顾一下基础知识
2013/01/15 Javascript
防止xss和sql注入:JS特殊字符过滤正则
2013/04/18 Javascript
jQuery图片滚动图片的效果(另类实现)
2013/06/02 Javascript
JS Jquery 遍历,筛选页面元素 自动完成(实现代码)
2013/07/08 Javascript
javascript 获取浏览器版本
2015/01/21 Javascript
jquery使用remove()方法删除指定class子元素
2015/03/26 Javascript
全面解析Bootstrap弹窗的实现方法
2015/12/01 Javascript
基于JavaScript实现网页倒计时自动跳转代码
2015/12/28 Javascript
Centos7 中安装 Node.js v4.4.4
2016/11/03 Javascript
Angular.js中定时器循环的3种方法总结
2017/04/27 Javascript
JS实现点击Radio动态更新table数据
2017/07/18 Javascript
VUE实现表单元素双向绑定(总结)
2017/08/08 Javascript
JS中Map和ForEach的区别
2018/02/05 Javascript
在Vue中使用Viser说明(基于AntV-G2可视化引擎)
2020/10/28 Javascript
[48:45]Ti4 循环赛第二日 NEWBEE vs EG
2014/07/11 DOTA
[02:44]重置世界,颠覆未来——DOTA2 7.23版本震撼上线
2019/12/01 DOTA
python中as用法实例分析
2015/04/30 Python
Python实现随机生成有效手机号码及身份证功能示例
2017/06/05 Python
Python3实现汉语转换为汉语拼音
2019/07/08 Python
Python 在OpenCV里实现仿射变换—坐标变换效果
2019/08/30 Python
python多进程使用函数封装实例
2020/05/02 Python
python ssh 执行shell命令的示例
2020/09/29 Python
建筑施工实习自我鉴定
2013/09/19 职场文书
总经理驾驶员岗位职责
2013/12/04 职场文书
会计人员演讲稿
2014/09/11 职场文书
无刑事犯罪记录证明范本
2014/09/29 职场文书
2014年班组建设工作总结
2014/12/01 职场文书
班级元旦晚会开幕词
2015/01/29 职场文书
幼儿教师师德培训心得体会
2016/01/09 职场文书
建立共青团委员会的请示
2019/04/02 职场文书