探索 Python 的 itertools 模块：释放迭代器的力量

Python 的 itertools 模块是标准库中用于高效处理迭代任务的强大工具包，通过提供排列组合、无限迭代、链接分组等功能，显著简化复杂迭代逻辑。以下是其核心功能与实际应用的详细解析：

• itertools.permutations(iterable, r=None)生成可迭代对象中所有长度为 r 的元素排列（顺序敏感）。示例：生成字符串 "ABC" 的所有 2 字符排列

  import itertoolsperms = itertools.permutations("ABC", 2)print(list(perms)) # 输出: [('A', 'B'), ('A', 'C'), ('B', 'A'), ('B', 'C'), ('C', 'A'), ('C', 'B')]

• itertools.combinations(iterable, r)生成所有长度为 r 的元素组合（顺序不敏感）。示例：生成字符串 "ABC" 的所有 2 字符组合

  combs = itertools.combinations("ABC", 2)print(list(combs)) # 输出: [('A', 'B'), ('A', 'C'), ('B', 'C')]

• itertools.count(start=0, step=1)生成无限递增序列，适用于模拟连续数值或动态数据流。示例：生成 0 到 5 的偶数序列

  counts = itertools.count(start=0, step=2)for i in range(3): print(next(counts)) # 输出: 0, 2, 4

• itertools.cycle(iterable)无限循环可迭代对象，常用于周期性任务（如轮询）。示例：循环输出 "AB" 3 次

  cycles = itertools.cycle("AB")for _ in range(3): print(next(cycles), end=" ") # 输出: A B A

• itertools.chain(*iterables)合并多个可迭代对象为一个连续序列，避免嵌套循环。示例：合并两个列表并遍历

  list1 = [1, 2]list2 = ["a", "b"]chained = itertools.chain(list1, list2)print(list(chained)) # 输出: [1, 2, 'a', 'b']

• itertools.groupby(iterable, key=None)根据键函数对连续元素分组，需先排序以保证分组正确性。示例：按字符串长度分组

  words = ["apple", "pear", "banana", "kiwi"]sorted_words = sorted(words, key=len)groups = itertools.groupby(sorted_words, key=len)for key, group in groups: print(f"{key}: {list(group)}") # 输出: 4: ['pear', 'kiwi'], 5: ['apple'], 6: ['banana']

使用排列生成所有可能的 3 位数字密码：

结合 count 和 takewhile 生成有限范围内的斐波那契数列：

使用 cycle 轮询分配任务到 3 个服务器：

• 内存效率：itertools 返回迭代器而非列表，适合处理大规模数据（如日志文件、数据库查询结果）。
• 惰性求值：仅在需要时生成值，避免一次性加载全部数据。
• 排序依赖：groupby 前需对数据按分组键排序，否则分组可能不完整。
• 无限迭代控制：使用 islice 或循环条件限制无限迭代器的输出范围。

• itertools.product：生成可迭代对象的笛卡尔积，替代嵌套循环。示例：计算两个列表的所有坐标组合x_coords = [1, 2]y_coords = ["A", "B"]for coord in itertools.product(x_coords, y_coords): print(coord) # 输出: (1, 'A'), (1, 'B'), (2, 'A'), (2, 'B')

通过灵活运用 itertools，开发者能以更简洁的代码实现复杂的迭代逻辑，同时提升程序性能和可读性。