在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)]