itertools 介绍

时间:2022-02-01 10:02:52

在python中itertool为python提供一系列迭代iterator的方法。

第一个:组合 排列

itertools.combinations(sq, r)

该函数的作用是在列表sq中穷举所有的组合项目。 地一个参数是列表,第二个参数一个项中的个数。

比如参数为‘abcd’和2。意思为从abcd中两两组合。 ab,ac,ad,bc,bd,cd

import itertools
str = 'abcd'
com = itertools.combination(str, 2)
print [item for item in com]

结果:[('a', 'b'), ('a', 'c'), ('a', 'd'), ('b', 'c'), ('b', 'd'), ('c', 'd')]

这是组合,自然有排列。

排列和组合的区别在于一个元素tuple内,先后顺序是否决定一个元素。比如(a,b)与(b,a),组合认为这是一个tuple,而排列认为这是两个。

组合的函数有:combinations,combinations_with_replacement。 两者区别在于是否出现如(a,a),(b,b)这种元素。

排列:product, permutations

itertools.compress(list, selector):

根据selector的true和false选择list中的字符

比如list=‘abcd' selector=[1,2,0,0] 选择结果ab

import itertools
test = 'abcdefghijk'
selector = [1,2,3,0,0,0,1,1,1,1,1]
it = itertools.compress(test, selector)
for [item for item in it]
print [item for item in it]

 结果:['a', 'b', 'c', 'g', 'h', 'i', 'j', 'k']

无限的函数

itertools.count(start, step)

从start开始,每次增长step。

如itertools.count(2, 5)  迭代后  2, 7, 12, 17, 22, 27,。。。。。

itertools.circle(p)

圈,很形象的函数名称。将p列表周而复始的迭代。

如itertools.circle('abcd'), 结果: a b c d a b c d a b c d a b c d......

itertools.repeat(elem, n)

将elem循环n次,如果没有参数n,无限循环。

itertools.repeat('abc', 4) 结果 abc abc abc abc

itertools.repeat('ab') 结果:ab ab ab 。。。。

根据条件选择内容:

itertools.dropwhile(pred, list)

itertools.takewhile(pred, list)

将list内容元素放入pred中进行判断,如果为True, dropwhile:丢弃, takewhile:保留。

import itertools
test = [3,4,5,6,7,8,9]
it = itertools.dropwhile(lambda x:x<6, test)
print [item for item in it]
it = itertools.takewhile(lambda x:x<6, test)
print [item for item in it]

 结果:[6, 7, 8, 9]    [3, 4, 5]

类似函数:ifilter, ifilterfalse

函数调用

itertools.imap(func, para1list, para2list,×)

从para1list和para2list获取参数调用函数func。

import itertools
def func(a, b, c):
return a + b + c
list1 = [1,2,3,4]
list2 = [5,6,7,8]
list3 = [0,1,0,1]
it = itertools.imap(func, list1, list2, list3)
print [item for item in it]

  结果:[6, 9, 10, 13]

类似函数:starmap

打包:

itertools.izip(list1, list2,*)

将list1中的元素与list2中的元素对应位置成为一个tuple。

list1 = [1,2,3,4]
list2 = [5,6,7,8]
list3 = [0,1,0,1]
it = itertools.izip(list1, list2, list3)
print [item for item in it]

  结果:[(1, 5, 0), (2, 6, 1), (3, 7, 0), (4, 8, 1)]

如果列表长度不一致,返回最短。

list1 = [1,2,3,4]
list2 = [5,6]
list3 = [0,1,0,1]
it = itertools.izip(list1, list2, list3)
print [item for item in it]

  结果:[(1, 5, 0), (2, 6, 1)]

如果希望根据最长的列表返回,这就需要对短的列表进行填充。

list1 = [1,2,3,4]
list2 = [5,6]
list3 = [0,1,0,1]
it = itertools.izip_longest(list1, list2, list3, fillvalue='x')
print [item for item in it]

  结果:[(1, 5, 0), (2, 6, 1), (3, 'x', 0), (4, 'x', 1)]