itertools是Python中高效处理迭代器的内置模块,提供内存友好的工具函数。1. 生成无限序列:count、cycle、repeat可创建无限迭代器;2. 有限迭代器:chain、islice、compress用于组合或筛选数据;3. 组合生成器:product、permutations、combinations等生成数学结构。其函数基于C实现,返回迭代器,节省内存,适用于大数据处理、参数组合等场景。例如combinations(['A','B','C'], 2)输出所有两字母组合,简化循环逻辑,提升代码效率。
itertools 是 Python 中用于处理迭代器的内置模块,它提供了一系列高效、内存友好的工具函数,用来创建、组合和操作迭代器。这些函数通常在处理大量数据或需要惰性求值(lazy evaluation)时非常有用,因为它们不会立即生成所有结果,而是按需产生元素。
1. 生成无限序列
这些函数可以生成无限长度的迭代器,适合与 break 或切片配合使用:
2. 有限迭代器
对已有可迭代对象进行变换或组合:
3. 组合生成器
用于生成排列、组合等数学结构:
itertools 函数基于 C 实现,运行速度快。更重要的是,它们返
回的是迭代器,不会一次性占用大量内存。例如,在遍历所有可能的配对时,使用 product 可以逐个获取结果,而不是先构建一个庞大的列表。
比如你想找出 ['A', 'B', 'C'] 中任意两个字母的组合:
from itertools import combinationsfor pair in combinations(['A','B','C'], 2): print(pair)# 输出: ('A','B'), ('A','C'), ('B','C')基本上就这些。它不复杂,但在写循环和数据处理时能大大简化代码逻辑。用好 itertools 能让你的 Python 更简洁高效。