Python itertools|极客教程

Python itertools

在 Python 编程语言中，itertools 模块提供了一组用于高效处理迭代对象的函数。通过使用 itertools，我们可以避免手动编写循环代码，并且可以节省时间和精力。这个模块包含了许多有用的函数，包括无限迭代器、组合生成器、排列生成器等等。在本篇文章中，我们将深入探讨 itertools 模块的各种使用方式。

1. 无限迭代器

itertools 中有一些函数可以用来生成无限的迭代器，比如 count()、cycle() 和 repeat()。下面是一些示例代码：

示例代码：

import itertools

# 生成从0开始的无限整数序列
infinite_counter = itertools.count()
for i in range(5):
    print(next(infinite_counter))

# 生成循环重复的元素
infinite_cycle = itertools.cycle('ABC')
for i in range(5):
    print(next(infinite_cycle))

# 重复元素无限次
infinite_repeat = itertools.repeat(10)
for i in range(5):
    print(next(infinite_repeat))

运行结果：

2. 组合生成器

itertools 中的 combinations() 函数可以用来生成给定元素集合的所有可能的组合。下面是一个示例代码：

示例代码：

import itertools

# 生成长度为2的所有组合
combinations = itertools.combinations('ABC', 2)
for combination in combinations:
    print(combination)

运行结果：

('A', 'B')
('A', 'C')
('B', 'C')

3. 排列生成器

itertools 中的 permutations() 函数可以用来生成给定元素集合的所有可能的排列。下面是一个示例代码：

示例代码：

import itertools

# 生成长度为2的所有排列
permutations = itertools.permutations('ABC', 2)
for permutation in permutations:
    print(permutation)

运行结果：

('A', 'B')
('A', 'C')
('B', 'A')
('B', 'C')
('C', 'A')
('C', 'B')

4. 笛卡尔积生成器

itertools 中的 product() 函数可以用来生成多个集合的笛卡尔积。下面是一个示例代码：

示例代码：

import itertools

# 生成两个集合的笛卡尔积
product = itertools.product('AB', '12')
for p in product:
    print(p)

运行结果：

('A', '1')
('A', '2')
('B', '1')
('B', '2')

5. 压缩生成器

itertools 中的 compress() 函数可以用来根据一个选择器序列来过滤一个可迭代对象。下面是一个示例代码：

示例代码：

import itertools

data = 'ABCDEF'
selectors = [True, False, True, False, True, False]

result = itertools.compress(data, selectors)
for item in result:
    print(item)