Python itertools模块

news/2024/10/29 21:58:47/

count 功能详解

count(start=0,step=1) 函数有两个参数,其中 step 是默认参数,可选的,默认值为 1。该函数返回一个新的迭代器,从 start 开始,返回以 step 为步长的均匀间隔的值。

import itertoolsx = itertools.count(1,2)for k in x:	print(k, end=", ")
# 输出结果如下 无穷无尽1, 3, 5, 7, 9, 11, 13, 15, ...

cycle 功能详解

cycle(iterable) 该函数会把接收到的序列无限重复下去。​​​​​​​

import itertoolsx = itertools.cycle("XYZ")for k in x:	print(k, end = ", ")  # 输出结果如下 无穷无尽X, Y, Z, X, Y, Z, X, Y, Z, ...

注意,该函数可能需要相当大的辅助空间(取决于 iterable 的长度)。

repeat 功能详解

repeat(object, times) 该函数创建一个迭代器,不断的重复 object,当然如果指定 times 的话,则只会重复 times 次。​​​​​​​

import itertoolsx = itertools.repeat("XYZ")for k in x:	print(k, end = ", ")  # 输出结果如下 无穷无尽XYZ, XYZ, XYZ, XYZ, XYZ, XYZ, ...​​​​​​
import itertoolsx = itertools.repeat("XYZ", 3)print(list(x))
# 输出结果如下 只会输出三次['XYZ', 'XYZ', 'XYZ']

注意:无限循环迭代器只有在 for 循环中才会不断的生成元素,如果只是创建一个迭代器对象,则不会事先生成无限个元素。

chain 功能详解

chain(*iterables) 该函数创建一个新的迭代器,会将参数中的所有迭代器全包含进去。​​​​​​​

import itertoolsx = itertools.chain("abc", "xyz")print(list(x))
# 输出结果如下['a', 'b', 'c', 'x', 'y', 'z']

groupby 功能详解

groupby(iterable, key=None) 分组函数,将 key 函数作用于序列的各个元素。根据 key 函数的返回值将拥有相同返回值的元素分到一个新的迭代器。类似于 SQL 中的 GROUP BY 操作,唯一不同的是该函数对序列的顺序有要求,因为当 key 函数的返回值改变时,迭代器就会生成一个新的分组。因此在使用该函数之前需要先使用同一个排序函数对该序列进行排序操作。​​​​​​​

import itertoolsdef sortBy(score):	if score > 80:		return "A"	elif score >= 60:		return "B"	else:		return "C"
scores = [81, 82, 84, 76, 64, 78, 59, 44, 55, 89]for m, n in itertools.groupby(scores, key=sortBy):	print(m, list(n))
# 输出结果如下A [81, 82, 84]B [76, 64, 78]C [59, 44, 55]A [89]

我们可以看到,该函数根据我们自定义的排序函数 sortBy 将列表中的元素进行了分组操作,只是我们发现最后一个怎么多了一个 A 的分组呢,这就是我们上面说所得「当 key 函数的返回值改变时,迭代器就会生成一个新的分组」。所以,我们需要事先对列表用 sortBy 函数排一下序。​​​​​​​

scores = [81, 82, 84, 76, 64, 78, 59, 44, 55, 89]scores = sorted(scores, key=sortBy)for m, n in itertools.groupby(scores, key=sortBy):	print(m, list(n))
# 输出结果如下A [81, 82, 84]B [76, 64, 78]C [59, 44, 55]A [89]

compress 功能详解

compress(data, selectors) 该函数功能很简单,就是根据 selectors 中的值判断是否保留 data 中对应位置的值。​​​​​​​

import itertoolsdata = [81, 82, 84, 76, 64, 78]tf = [1,1,0,1,1,0]print(list(itertools.compress(data, tf)))
# 输出结果如下[81, 82, 76, 64]

dropwhile 功能详解

dropwhile(predicate, iterable) 创建一个迭代器,从 predicate 首次为 false 时开始迭代元素。​​​​​​​

import itertoolsx = itertools.dropwhile(lambda x: x < 5, [1,3,5,7,4,2,1])print(list(x))
# 输出结果如下[5, 7, 4, 2, 1]

由以上得知,即使 predicate 首次为 false 后面的元素不满足 predicate 也同样会被迭代。

filterfalse 功能详解

filterfalse(predicate, iterable) 创建一个迭代器,返回 iterable 中 predicate 为 false 的元素。​​​​​​​

import itertoolsx = itertools.filterfalse(lambda x: x < 5, [1,3,5,7,4,2,1])print(list(x))
# 输出结果如下[5, 7]

islice 功能详解

islice(iterable, start, stop[, step]) 对 iterable 进行切片操作。从 start 开始到 stop 截止,同时支持以步长为 step 的跳跃。​​​​​​​

import itertoolsprint(list(itertools.islice('123456789', 2)))print(list(itertools.islice('123456789', 2, 4)))print(list(itertools.islice('123456789', 2, None)))print(list(itertools.islice('123456789', 0, None, 2)))
# 输出结果如下['1', '2']['3', '4']['3', '4', '5', '6', '7', '8', '9']['1', '3', '5', '7', '9']

starmap 功能详解

starmap(function, iterable) 从可迭代对象中获取参数来执行该函数。​​​​​​​

import itertoolsprint(list(itertools.starmap(pow,[(2,10), (3,3)])))
# 输出结果如下[1024, 27]

takewhile 功能详解

takewhile(predicate, iterable) 创建一个迭代器,遇到 predicate 为 false 则停止迭代元素。与 dropwhile 完全相反。​​​​​​​

import itertoolsx = itertools.takewhile(lambda x: x < 5, [1,3,5,7,4,2,1])print(list(x))
# 输出结果如下[1, 3]

product 功能详解

product(*iterables, repeat=1) 输出可迭代对象的笛卡尔积,有点类似于嵌套循环。其中 repeat 可以设置循环次数。​​​​​​​

import itertoolsprint(list(itertools.product("ab", "12")))print(list(itertools.product("ab", "ab")))print(list(itertools.product("ab", repeat=2)))
# 输出结果如下[('a', '1'), ('a', '2'), ('b', '1'), ('b', '2')][('a', 'a'), ('a', 'b'), ('b', 'a'), ('b', 'b')][('a', 'a'), ('a', 'b'), ('b', 'a'), ('b', 'b')]

permutations 功能详解

permutations(iterable, r=None) 返回 iterable 中长度为 r 的所有排列。默认值 r 为 iterable 的长度。即使元素的值相同,不同位置的元素也被认为是不同的。​​​​​​​

import itertoolsprint(list(itertools.permutations("aba", r=2)))
# 输出结果如下[('a', 'b'), ('a', 'a'), ('b', 'a'), ('b', 'a'), ('a', 'a'), ('a', 'b')]

combinations 功能详解

combinations(iterable, r=None) 返回 iterable 中长度为 r 的有序排列。默认值 r 为 iterable 的长度。与 permutations 操作不同的是该函数严格按照 iterable 中元素的顺序进行排列。​​​​​​​

import itertoolsprint(list(itertools.combinations("abc", r=2)))
# 输出结果如下[('a', 'b'), ('a', 'c'), ('b', 'c')]

combinations_with_replacement 功能详解

combinations_with_replacement(iterable, r=None) 返回 iterable 中长度为 r 的有序排列。默认值 r 为 iterable 的长度。与 combinations 操作不同的是该函数允许每个元素重复出现。​​​​​​​

import itertoolsprint(list(itertools.combinations_with_replacement("abc", r=2)))
# 输出结果如下[('a', 'a'), ('a', 'b'), ('a', 'c'), ('b', 'b'), ('b', 'c'), ('c', 'c')]

itertools 总结

本文总结了 itertools 模块的常规操作,学习并掌握这些极为便利的操作非常有助于提高自己的编码效率。


http://www.ppmy.cn/news/88918.html

相关文章

JSP+sql实验教学管理系统(系统+论文+开题报告+封面+中期检查表+英文文献)

现在的实验教学网络管理系统存在实验题目分发慢,文档交换不方便,师生交流困难等方面的问题。为了解决现有系统的这些问题,特开发新的实验教学网络管理系统。 作者以实际应用为开发背景,运用软件工程原理和开发方法,采用当前网络开发主流技术,结合JSP和SQL数据库技术,设…

设计事务所项目管理指南

在数字化的浪潮下&#xff0c;各行各业都面临着升级转型的问题。对设计团队而言&#xff0c;传统的管理方式已经无法满足日益前进的团队需求。 设计事务所可能存在的管理问题&#xff1a; 1&#xff0c;项目过程中信息流通慢&#xff0c;成员工作进度无法及时同步&#xff1b; …

springboot+vue学生综合测评系统(java项目源码+文档)

风定落花生&#xff0c;歌声逐流水&#xff0c;大家好我是风歌&#xff0c;混迹在java圈的辛苦码农。今天要和大家聊的是一款基于springboot的学生综合测评系统。项目源码以及部署相关请联系风歌&#xff0c;文末附上联系信息 。 &#x1f495;&#x1f495;作者&#xff1a;风…

四川师范大学何云:事项法会计从五大方面助力企业创造价值

近日&#xff0c;由用友主办的「智能会计 价值财务」2023企业数智化财务创新峰会收官站圆满举办。来自知名院校的专家学者、央国企等大型企业财务领路人相约成都&#xff0c;一同见证“智能会计”新时代的到来&#xff0c;并肩探讨“价值财务”新主张。 为更好传递智能会计对企…

iOS-最全的App上架教程

App上架教程 在上架App之前想要进行真机测试的同学&#xff0c;请查看《iOS- 最全的真机测试教程》&#xff0c;里面包含如何让多台电脑同时上架App和真机调试。 P12文件的使用详解 注意&#xff1a; 同样可以在Build Setting 的sign中设置证书&#xff0c;但是有点麻烦&…

如何使用 Fail2ban 防止对 Linux 的暴力攻击?

在当今数字化世界中&#xff0c;网络安全成为了一个极其重要的话题。Linux 作为一种广泛使用的操作系统&#xff0c;也面临着各种网络攻击的风险&#xff0c;包括暴力攻击、密码破解和恶意登录等。为了保护 Linux 系统的安全&#xff0c;我们可以使用 Fail2ban 这样的工具来防止…

航空公司预订票数学建模论文

航空公司预订票数学建模论文篇1 试谈机票订票模型与求解 一、概述 1. 问题背景描述 在激烈的市场竞争中&#xff0c;航空公司为争取更多的客源而开展的一个优质服务项目是预订票业务,本模型针对预订票业务&#xff0c;建立二元规划订票方案&#xff0c;既考虑航空公司的利润最大…

mysql 库的操作

文章目录 mysql 库的操作1. 创建数据库创建数据库案例 2. 字符集和校验规则查看系统默认的字符集合校验规则查看数据库支持的字符集查看数据库支持的字符集较验规则校验规则对数据库的影响 3. 操作数据库查看数据库显示创建语句修改数据库删除数据库查看数据库连接情况 mysql 库…