Python里itertools.chain用法 迭代器工具itertools中chain功能解析

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 可以让你在处理多个序列时更加灵活和高效。

以上就是Python里itertools.chain用法 迭代器工具itertools中chain功能解析的详细内容，更多请关注php中文网其它相关文章！