最直接写法是[item for sublist in nested_list for item in sublist],外层循环遍历子列表、内层遍历元素,顺序不可颠倒;该写法要求所有子项均为可迭代对象,不支持混合类型或任意深度嵌套。

用列表推导式扁平化两层嵌套列表最直接写法
对形如 [[1, 2], [3, 4, 5], [6]] 这样的两层嵌套,列表推导式是语法最紧凑、可读性最好的解法。核心是把外层循环和内层循环都写在同一个方括号里,顺序不能颠倒。
- 外层
for sublist in nested_list遍历每个子列表 - 内层
for item in sublist遍历子列表里的每个元素 - 顺序必须是
[item for sublist in nested_list for item in sublist],反过来会报错
示例:
data = [[1, 2], [3, 4, 5], [6]] flat = [item for sublist in data for item in sublist] # 结果:[1, 2, 3, 4, 5, 6]
遇到空子列表或非列表元素时会出错
上面的写法默认所有 sublist 都是可迭代对象。如果嵌套结构不规整——比如含 None、数字、字符串,或空列表——for item in sublist 会触发 TypeError: 'int' object is not iterable 或类似错误。
- 空列表本身没问题(
for x in []安静跳过) - 但
[1, [2, 3], 4]这种混合类型不行,因为1和4不可迭代 - 若需容错,得先加类型判断,此时列表推导式会变臃肿,不如改用函数封装
三层及以上嵌套不能靠简单叠加 for
列表推导式只适合明确层数的扁平化。[[[1, 2]], [[3]], [[4, 5, 6]]] 是三层,写成 [x for a in l for b in a for x in b] 虽然语法合法,但可读性骤降,且极易漏掉某一层变量名或顺序错乱。
- 每多一层,推导式长度和出错概率指数级上升
- 没有通用“自动展开 N 层”的列表推导式写法
- 真要处理任意深度,递归函数或
itertools.chain.from_iterable()套用更可靠
性能和内存占用其实和 for 循环几乎一样
别被“推导式更快”的传言误导。CPython 下,[item for ...] 和等价的显式 for 循环生成同样字节码,速度差异可忽略。但要注意:
- 它是一次性生成完整新列表,不支持惰性求值;大数据量时内存压力和普通
for一样 - 如果只需要逐个处理元素,用生成器表达式
(item for sublist in data for item in sublist)更省内存 - 涉及字符串时小心:
for char in "abc"会拆成字符,可能不是你想要的“扁平化”语义
实际项目里,两层嵌套用列表推导式很自然;一旦出现类型不一致、深度不确定或需要中途过滤/转换,就该果断换函数封装——推导式不是银弹,它只在边界清晰时才真正省事。
Python免费学习笔记(深入):立即使用
在学习笔记中,你将探索 Python 的核心概念和高级技巧!











