概念:
组合迭代器:对序列的排列组合,求序列的笛卡尔积,python itertools提供了4中组合迭代器方法。
案例:
1、itertools.product(*iterables,repeat=1):
求笛卡尔积,相当于嵌套的for循环,如product(A,B)其实就是等于((x,y) for x in A for y in B ))。
每次迭代时将最右侧的元素向后迭代。
repeat设置重复迭代次数,默认是1,例如product(A,repeat=4) 等于product(A,A,A,A)。
代码:
#itertools.product()
print (list(itertools.product('AB','xy')))
print (list((x,y) for x in 'AB' for y in 'xy'))
输出:
[('A', 'x'), ('A', 'y'), ('B', 'x'), ('B', 'y')]
[('A', 'x'), ('A', 'y'), ('B', 'x'), ('B', 'y')]
代码:
#itertools.product()
print (list(itertools.product('AB','xy',repeat=2)))
print(list(itertools.product(range(2),repeat=3)))
输出:
[('A', 'x', 'A', 'x'), ('A', 'x', 'A', 'y'), ('A', 'x', 'B', 'x'), ('A', 'x', 'B', 'y'),
('A', 'y', 'A', 'x'), ('A', 'y', 'A', 'y'), ('A', 'y', 'B', 'x'), ('A', 'y', 'B', 'y'),
('B', 'x', 'A', 'x'), ('B', 'x', 'A', 'y'), ('B', 'x', 'B', 'x'), ('B', 'x', 'B', 'y'),
('B', 'y', 'A', 'x'), ('B', 'y', 'A', 'y'), ('B', 'y', 'B', 'x'), ('B', 'y', 'B', 'y')]
[(0, 0, 0), (0, 0, 1), (0, 1, 0), (0, 1, 1), (1, 0, 0), (1, 0, 1), (1, 1, 0), (1, 1, 1)]
代码:
#itertools.product()
a = (1,2,3)
b = ('a','b','c')
c = itertools.product(a,b)
for i in c:
print (i)
输出:
(1, 'a')
(1, 'b')
(1, 'c')
(2, 'a')
(2, 'b')
(2, 'c')
(3, 'a')
(3, 'b')
(3, 'c')
代码:
#itertools.product()
print (list(itertools.product((0,1),(0,1),(0,1))))
print (list(itertools.product('ab',repeat=2)))
输出:
[(0, 0, 0), (0, 0, 1), (0, 1, 0), (0, 1, 1), (1, 0, 0), (1, 0, 1), (1, 1, 0), (1, 1, 1)]
[('a', 'a'), ('a', 'b'), ('b', 'a'), ('b', 'b')]
代码:
#itertools.product()
for i,j in itertools.product(range(2),range(2)):
print ('i = %s' %i)
print ('j = %s' %j)
for x in range(2):
for y in range(2):
print ('x = %s'% x,'y = %s'% y)
输出:
i = 0
j = 0
i = 0
j = 1
i = 1
j = 0
i = 1
j = 1
x = 0 y = 0
x = 0 y = 1
x = 1 y = 0
x = 1 y = 1
2、itertools.permutations(iterable,r=None):
连续返回由iterable元素生成长度为r的排列。
如果r=None或者省略,则返回的序列长度与iterable中的项目数量相同。
元素的值相同,位置不同也认为这些元素是不同的。
#itertools.permutations()
print (list(itertools.permutations('ABCD',2)))
print (list(itertools.permutations(range(3),3)))
结果:
[('A', 'B'), ('A', 'C'), ('A', 'D'), ('B', 'A'), ('B', 'C'), ('B', 'D'), ('C', 'A'), ('C', 'B'), ('C', 'D'), ('D', 'A'), ('D', 'B'), ('D', 'C')]
[(0, 1, 2), (0, 2, 1), (1, 0, 2), (1, 2, 0), (2, 0, 1), (2, 1, 0)]
3、itertools.combinations(iterable,r):
返回由itertable组成长度为r的子序列。
子序列的项由输入的iterable中的顺序排序(不带重复)
#itertools.combinations()
print (list(itertools.combinations('ABCD',2)))
print (list(itertools.combinations(range(3),3)))
输出:
[('A', 'B'), ('A', 'C'), ('A', 'D'), ('B', 'C'), ('B', 'D'), ('C', 'D')]
[(0, 1, 2)]
4、itertools.combinations_with_replacement(iterable,r)
返回由itertable组成长度为r的子序列。
子序列的项由输入的iterable中的顺序排序(带重复)
print (list(itertools.combinations_with_replacement('ABCD',2)))
结果:
[('A', 'A'), ('A', 'B'), ('A', 'C'), ('A', 'D'), ('B', 'B'), ('B', 'C'), ('B', 'D'), ('C', 'C'), ('C', 'D'), ('D', 'D')]