itertools.chain(*nested) 是展开双层嵌套列表最省事的方法,不建中间列表、不递归、纯流式拼接,但需解包且要求子项为非字符串可迭代对象。

用 itertools.chain 展开双层嵌套列表最省事
它不建中间列表、不递归、不改原结构,纯流式拼接,适合大列表或生成器场景。比双层 for 循环写法少两行,也更易读。
常见错误是直接传入嵌套列表给 chain(),结果只展开一层——chain(nested) 会把每个子列表当一个元素,必须用解包:chain(*nested)。
- 适用场景:已知只有两层、子列表非空、不需要索引或条件过滤
- 如果嵌套里有
None或非迭代对象,chain(*nested)会报TypeError: 'NoneType' object is not iterable - 兼容性无问题,Python 2.6+ 都支持
from itertools import chain nested = [[1, 2], [3, 4], [5]] flat = list(chain(*nested)) # → [1, 2, 3, 4, 5]
双层 for 循环最直白,但容易写错顺序
很多人写成 for i in nested: for j in i 却忘了把 j 收集起来,或者漏了初始化空列表,导致结果为空或变量未定义。
它灵活,能加判断、转换、跳过空子列表,但性能略逊于 chain(多一次循环变量绑定和作用域查找)。
立即学习“Python免费学习笔记(深入)”;
- 写法必须是
for sublist in nested: for item in sublist:,顺序反了就遍历错维度 - 如果某子列表是
None,运行时抛TypeError: 'NoneType' object is not iterable - 想保留原列表中空列表的占位,得显式处理:
if sublist is not None
nested = [[1, 2], None, [3]]
flat = []
for sublist in nested:
if sublist: # 跳过 None 和空列表
for item in sublist:
flat.append(item)遇到空子列表或 None 怎么安全展开
不管是 chain(*nested) 还是双层 for,遇到 None 或字符串等非列表类型都会崩。不能靠“试试看”,得提前筛或兜底。
最稳的方式是统一转成可迭代对象再处理,比如用 iter(x) if x else iter([]),但更常用的是列表推导式配合 isinstance 判断。
- 别用
if sublist:判空——它会把[0]、[False]当作假值误删 - 正确判子列表是否可迭代且非字符串:
hasattr(sublist, '__iter__') and not isinstance(sublist, str) - 如果数据来源不可控(比如 JSON 解析结果),优先用
try/except包裹内层循环
from itertools import chain nested = [[1, 2], "oops", None, [3]] safe_nested = [x for x in nested if hasattr(x, '__iter__') and not isinstance(x, str)] flat = list(chain(*safe_nested)) # → [1, 2, 3]
sum(nested, []) 看似简洁,但千万别在生产环境用
它利用列表的 + 是左结合的特性,把所有子列表累加起来。初看很酷,实则隐患极大:时间复杂度 O(n²),因为每次 + 都要复制前面所有元素。
1000 个长度为 10 的子列表,就要做约 500 万次元素拷贝。而 chain 是 O(n),双层 for 是 O(n) 且常数更小。
- 只适用于玩具数据或临时调试,绝对不要出现在处理日志、API 响应、数据库结果的代码里
- 如果子列表里有元组或其它类型,
sum直接报TypeError: can only concatenate list (not "tuple") to list - Python 3.12+ 对这种用法已有警告提示
# 千万别这么写 nested = [[1], [2], [3]] flat = sum(nested, []) # → [1, 2, 3],但慢、脆、难维护
真正麻烦的不是“怎么展开”,而是嵌套结构本身是否稳定——比如从 API 拿到的数据,可能某次返回 [{"a":1}, {"a":2}],下次变成 {"data": [{"a":1}, {"a":2}]}。这时候硬写展开逻辑不如先统一数据契约。










