您好,欢迎来到三六零分类信息网!老站,搜索引擎当天收录,欢迎发信息

深入认识Python中的itertools模块

2024/3/12 16:30:22发布20次查看
python视频教程栏目介绍itertools模块。
在python中有一个功能强大的迭代工具包itertools,是python自带的标准工具包之一。
product由于itertools是内置库,不需要任何安装,直接import itertools即可。
product 用于求多个可迭代对象的笛卡尔积(cartesian product),它跟嵌套的 for 循环等价.即:
笛卡尔乘积是指在数学中,两个集合x和y的笛卡尔积(cartesian product),又称直积,表示为x × y。
product(a, b)和 ``((x,y) for x in a for y in b)`一样.
import itertoolsfor item in itertools.product([1,2,3],[100,200]):    print(item)        # 输出如下(1, 100)(1, 200)(2, 100)(2, 200)(3, 100)(3, 200)复制代码
permutations通俗地讲,permutations就是返回可迭代对象的所有数学或者字符的全排列方式。
全排列,即产生指定数目的元素的所有排列(顺序有关),也就是高中排列组合中的那个a。
permutations它接受一个集合对象,然后产生一个元组序列。
比如print(list(itertools.permutations('abc',3))),共有a33=6a_3^3=6a33=6种情况。
items = ['a','b','c']from itertools import permutationsfor i in permutations(items):    print(i) #排列组合print(list(itertools.permutations('abc',3))) # 输出如下('a', 'b', 'c')('a', 'c', 'b')('b', 'a', 'c')('b', 'c', 'a')('c', 'a', 'b')('c', 'b', 'a')[('a', 'b', 'c'), ('a', 'c', 'b'), ('b', 'a', 'c'), ('b', 'c', 'a'), ('c', 'a', 'b'), ('c', 'b', 'a')]复制代码
如果需要指定长度的所有排列,可以传递一个可选的长度参数r。
items = ['a','b','c']from itertools import permutationsfor i in permutations(items,2):    print(i) #排列组合    # 输出如下('a', 'b')('a', 'c')('b', 'a')('b', 'c')('c', 'a')('c', 'b')复制代码
combinations求列表或生成器中指定数目的元素不重复的所有组合
itertools.permutations(iter,r) 和 itertools.combinations(iter,r)的区别是:前者是permutations允许重复使用,后者combinations是不能重复使用
>>> print(list(itertools.combinations('abc',3)))[('a', 'b', 'c')]复制代码
combinations_with_replacementcombinations_with_replacement和combinations很相似,唯一的不同在于前者combinations_with_replacement集合类型中的数据是可以重复的
>>> print(list(itertools.combinations_with_replacement('abc',3)))[('a', 'a', 'a'), ('a', 'a', 'b'), ('a', 'a', 'c'), ('a', 'b', 'b'), ('a', 'b', 'c'), ('a', 'c', 'c'), ('b', 'b', 'b'), ('b', 'b', 'c'), ('b', 'c', 'c'), ('c', 'c', 'c')]复制代码
accumulateaccumulate用于对列表中元素逐个累加
>>> import itertools>>> x = itertools.accumulate(range(10))>>> print(list(x))[0, 1, 3, 6, 10, 15, 21, 28, 36, 45]复制代码
compresscompress()是筛选工具,它接受一个可迭代对象以及一个布尔选择序列作为输入,输出时会将所有布尔序列中为true的可迭代对象输出。
import itertoolsits=[a,b,c,d,e,f,g,h]selector=[true,false,1,0,3,false,-2,y]for item in itertools.compress(its,selector):    print (item)    acegh   复制代码
countcount(初值=0, 步长=1)是 创建一个迭代器,从传入的起始参数开始的均匀间隔的数值。
我们来看一个简单的例子
from itertools import countfor i in count(10): #从10开始无限循环    if i > 20:         break    else:        print(i)1011121314151617181920复制代码
chainchain链条,主要用来把多个序列连在一起做迭代。
import itertoolschain = itertools.chain([1, 2, 3], [4, 5, 6])for c in chain:   print(c)123456  复制代码
chain还有一个非常重要的功能就是展平列表。
>>> list(itertools.chain([1, 2, 3], [4, 5], [6] ,[7,8]))[1, 2, 3, 4, 5, 6, 7, 8]复制代码
cycleimport itertoolscycle = itertools.cycle([1, 2, 3])for c in cycle:   print(c)复制代码
运行结果输出 1 2 3 1 2 3……一直周而复始,永不停息。
相关免费学习推荐:python视频教程
以上就是深入认识python中的itertools模块的详细内容。
该用户其它信息

VIP推荐

免费发布信息,免费发布B2B信息网站平台 - 三六零分类信息网 沪ICP备09012988号-2
企业名录 Product