python itertools 模块

itertools 模块的官方解释： Functional tools for creating and using iterators. （用于创建和使用迭代器的函数工具）

1.   product(*iterables, repeat=1) 

用于求多个可迭代对象的笛卡尔积(Cartesian Product)，它跟嵌套的 for 循环等价.即:  product(A, B) 和 ((x,y) for x in A for y in B)一样.

例：

'''
第一个参数是可迭代器
第二个参数的值是组合的长度
'''
from itertools import product
for i in (x for x in product('ABCD', repeat=2)):
    print(i)

# 输出结果
'''
('A', 'A')
('A', 'B')
('A', 'C')
('A', 'D')
('B', 'A')
('B', 'B')
('B', 'C')
('B', 'D')
('C', 'A')
('C', 'B')
('C', 'C')
('C', 'D')
('D', 'A')
('D', 'B')
('D', 'C')
('D', 'D')
'''


for i in (x for x in product(range(3), repeat=3)):
    print(i)

# 输出结果
'''
(0, 0, 0)
(0, 0, 1)
(0, 0, 2)
(0, 1, 0)
(0, 1, 1)
(0, 1, 2)
(0, 2, 0)
(0, 2, 1)
(0, 2, 2)
(1, 0, 0)
(1, 0, 1)
(1, 0, 2)
(1, 1, 0)
(1, 1, 1)
(1, 1, 2)
(1, 2, 0)
(1, 2, 1)
(1, 2, 2)
(2, 0, 0)
(2, 0, 1)
(2, 0, 2)
(2, 1, 0)
(2, 1, 1)
(2, 1, 2)
(2, 2, 0)
(2, 2, 1)
(2, 2, 2)
'''

2.  permutations(iterable, r=None)

在不存在重复元素的情况下生成全排列

例：

from itertools import permutations
for i in (x for x in permutations('ABCD', r=2)):
    print(i)
# 输出结果
'''
('A', 'B')
('A', 'C')
('A', 'D')
('B', 'A')
('B', 'C')
('B', 'D')
('C', 'A')
('C', 'B')
('C', 'D')
('D', 'A')
('D', 'B')
('D', 'C')
'''
for i in (x for x in permutations(range(3), r=3)):
    print(i)
# 输出结果
'''
(0, 1, 2)
(0, 2, 1)
(1, 0, 2)
(1, 2, 0)
(2, 0, 1)
(2, 1, 0)
'''

3.    cycle(iterable)

无限循环iterable中的元素

例：

from itertools import cycle

n = 0
for i in cycle('abcde'):
    print(i)
    n +=1

    if n >10:
        break
# 输出结果
'''
a
b
c
d
e
a
b
c
d
e
a
'''

4.    islice（iterable, stop)

迭代多少次后停止，stop为迭代次数

例：

from itertools import islice,cycle
for i in islice(cycle('abcde'),5):
    print(i)
# 输出结果
'''
a
b
c
d
e
'''