Python单行代码技巧：itertools高效用法

本文揭秘了如何利用 Python 的 itertools 模块与生成器表达式、列表推导式等协同工作，在单行代码中高效完成嵌套列表扁平化、元素组合生成、自增索引添加及无限循环取值等常见数据处理任务，既避免冗长的 for 循环，又保持代码简洁可读——掌握这些技巧，能让你的 Python 代码更精炼、更具函数式编程魅力。

在 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 但更灵活：

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 触发计算。不复杂但容易忽略。

以上就是本文的全部内容了，是否有顺利帮助你解决问题？若是能给你带来学习上的帮助，请大家多多支持golang学习网！更多关于文章的相关知识，也可关注golang学习网公众号。