本文共 827 字，大约阅读时间需要 2 分钟。

本文来源于公众号“python”，仅用于学术分享，侵权删，干货满满。

原文链接：

itertools模块是Python标准库中的一个强大工具箱，专为处理迭代器提供了高效的函数支持。通过itertools模块，可以简化繁琐的循环和迭代操作，显著提升代码效率。该模块内置多种功能，包括排列组合、无限迭代器、分组、过滤等，特别适用于数据处理、序列操作等场景。

1 无限迭代器

itertools模块提供了多种无限迭代器工具，适用于需要生成无限序列的场景。其中，三种主要的无限迭代器工具分别是count、cycle和repeat。

1.1 count：生成等差数列

count(start, step)函数可以从指定的start值开始，按固定step递增生成无限等差数列。

以下示例展示了如何生成从10开始，每次递增5的无限数列：

import itertools for num in itertools.count(10, 5):     if num > 30:         break     print(num)

运行该代码将输出：10, 15, 20, 25, 30

1.2 cycle：循环遍历序列

cycle(iterable)函数会将指定的可迭代对象无限循环遍历，适用于需要反复访问特定元素的场景。

以下示例展示了如何无限循环一个颜色列表的元素：

colors = ['red', 'green', 'blue'] import itertools cycled_colors = itertools.cycle(colors) for _ in range(6):     print(next(cycled_colors))

运行该代码将输出：red, green, blue, red, green, blue

1.3 repeat：重复生成序列

repeat函数可以将指定的可迭代对象无限重复生成，适用于需要多次重复相同序列的场景。

转载地址：http://dkofk.baihongyu.com/