itertools结合生成器可在单行高效处理可迭代对象:1. chain.from_iterable扁平化嵌套列表;2. combinations生成元素组合;3. count配合zip添加自增索引;4. cycle创建无限循环迭代器,通过next控制取值,利用list或推导式触发计算,实现简洁的单行操作。

在 Python3 中,itertools 本身不直接用于“单行处理代码”,但可以配合生成器表达式、map、zip 等工具,在一行内高效处理可迭代对象。常见用法是结合 itertools 提供的工具函数,在单行中完成循环、组合、过滤等操作,避免写多行 for 循环。
1. 使用 itertools.chain 扁平化多个列表
将多个列表合并为一个可迭代对象,适合处理嵌套结构:
from itertools import chain result = list(chain.from_iterable([[1, 2], [3, 4], [5]])) # result: [1, 2, 3, 4, 5]
2. 用 itertools.combinations 生成组合
一行内获取所有两两组合:
from itertools import combinations
pairs = list(combinations('ABC', 2))
# pairs: [('A', 'B'), ('A', 'C'), ('B', 'C')]
3. 利用 itertools.count 实现自增索引
配合 zip 给元素加序号,类似 enumerate 但更灵活:
立即学习“Python免费学习笔记(深入)”;
from itertools import count items = [x for x in zip(count(1), ['a', 'b', 'c'])] # items: [(1, 'a'), (2, 'b'), (3, 'c')]
4. 用 itertools.cycle 无限循环取值
常用于交替取值场景,配合 next 控制输出:
from itertools import cycle colors = cycle(['red', 'green']) take_three = [next(colors) for _ in range(3)] # take_three: ['red', 'green', 'red']
基本上就这些。通过把逻辑压缩进生成器或列表推导,再结合 itertools 工具,就能写出简洁高效的单行处理语句。关键是理解每个函数返回的是迭代器,需用 list() 或 for 触发计算。不复杂但容易忽略。











