itertools.chain 是 python 中用于连接多个可迭代对象的工具，返回一个按需生成元素的迭代器。它有两种主要用法：1. chain(iterable1, iterable2, …) 可依次连接多个可迭代对象，适用于合并多个列表或字符串；2. chain.from_iterable(iterable_of_iterables) 用于展平嵌套结构，如二维列表。常见应用场景包括处理日志文件、组合不同来源的数据以及减少数据拷贝。使用时需注意 chain 返回的是单次迭代器，无法重复遍历，且 from_iterable 要求每个元素都必须是可迭代的。
在 python 的 itertools 模块中，chain 是一个非常实用的工具，用来将多个可迭代对象连接在一起，形成一个连续的迭代器。它不会一次性把所有数据加载到内存中，而是按需生成元素，非常适合处理大体量的数据。
什么是 itertools.chain
itertools.chain 的作用是将多个可迭代对象（比如列表、元组、字符串等）依次连接起来，返回一个可以逐个访问这些元素的迭代器。它不改变原始数据，也不会复制数据，只是“按顺序读取”的一种方式。
使用方式主要有两种：
- chain(iterable1, iterable2, …)：直接传入多个可迭代对象
- chain.from_iterable(iterable_of_iterables)：传入一个包含多个可迭代对象的可迭代对象（比如列表的列表）
常见用法与示例
合并多个列表或字符串
这是最基础的用法。如果你有几个列表或者字符串，想一个个地遍历，而不是嵌套循环，就可以用 chain。
立即学习“Python免费学习笔记（深入）”；
from itertools import chain list1 = [1, 2, 3] list2 = ['a', 'b', 'c'] result = chain(list1, list2) for item in result: print(item)
输出：
1 2 3 a b c
这种方式比先合并成一个新列表再遍历更节省内存，尤其是处理大列表时。
使用 from_iterable 展开嵌套结构
当你有一个嵌套结构，比如二维列表，想要把它展平成一维来遍历，这时候可以用 chain.from_iterable()。
nested_list = [[1, 2], [3, 4], [5]] flattened = chain.from_iterable(nested_list) print(list(flattened)) # 输出: [1, 2, 3, 4, 5]
这个方法在处理 json 数据、网页爬虫结果等嵌套结构时非常有用。
在实际项目中的应用建议
- 处理日志文件：如果你有多个日志文件，每个文件一行行记录，可以用 chain 把它们串起来统一处理。
- 组合不同来源的数据：比如从数据库、API 和本地缓存各取出一部分数据，可以用 chain 把它们拼接后统一分析。
- 减少不必要的拷贝：比起用 + 运算符合并两个列表，chain 更高效，特别是当数据量大的时候。
一些注意事项：
- chain 返回的是一个迭代器，不是列表，不能重复遍历。
- 如果你需要多次使用结果，记得转成列表保存下来。
- from_iterable 要求输入的每个元素都必须是可迭代的，否则会报错。
基本上就这些。掌握好 itertools.chain 可以让你在处理多个序列时更加灵活和高效。
