目录
简介
itertools,是python的一个内置模块,功能强大,主要用于高效循环创建迭代器。注意一点,他返回的不是list,而是iterator
无穷迭代器
count()
count(start, [step])会创建一个无限迭代器,从当前数字无限循环下去,停止可以按Ctrl+c键

cycle()
cycle( p)会把传入的序列无限的循环打印

repeat()
repeat(elem [,n])可以把一个元素无限循环

也可以添加第二个参数来限制个数

有限迭代器
因为有限迭代器有十多个,这里只列举几个常用的函数
chain()
chain(p,q…)可以把一组迭代对象串联起来,形成一个更大的迭代器

groupby()
groupby(iterable,key=None])把迭代器中相邻的元素按照key函数分组,当key=None时,把相邻的重复元素进行分组


accumulate()
accumulate(iterable [,func])是一个计算迭代器,如果不指定参数函数,将会采取默认的求和函数

也可以指定参数函数,比如求最大值

关于有限迭代器还有更多有趣的函数,可以去官方文档查看
itertools官方文档
组合迭代器
组合操作在算法中经常会用到,因此这里就体现出来itertools模块的便捷性
下面来看看四个组合迭代器的用法
product()
product(p,q,…[repeat=1])得到的是可迭代的笛卡尔积,多个可迭代对象的笛卡尔积可以通过for循环来实现,比如product(a,b)用for循环可以表示为
((i,j) for i in a for j in b)

函数中的repeat参数表示前面可迭代序列的重复次数
比如product(a, repeat=4) 就表示 product(a, a, a, a)
permutations()
permutations(p[, r])返回的是一个长度为 r 的所有可能排列,无重复元素

第二个参数默认为迭代序列长度
combinations()
combinations(p, r)返回的是一个长度为r的组合,它是有序的,无重复元素

combinations_with_replacement()
combinations_with_replacement(p, r)返回的是一个长度为r的组合,它是有序的,元素可以重复

Python的itertools模块提供了高效的迭代器工具,包括无限迭代如count(), cycle(), repeat(),以及有限迭代器如chain(), groupby(), accumulate()等。此外,它还包含组合迭代器如product(), permutations(), combinations(), combinations_with_replacement(),用于创建笛卡尔积、排列和组合。这些函数在算法和数据处理中非常实用。
611

被折叠的 条评论
为什么被折叠?



