答案是使用列表推导式可将嵌套for循环简化为单行,如[(i, j) for i in range(3) for j in range(2)],支持条件过滤和多层嵌套,但应避免过度嵌套以保持可读性。

在 Python 中,将嵌套 for 循环写成单行通常可以通过列表推导式(List Comprehension)实现。这种方式不仅简洁,而且在多数情况下性能更好。
基本思路:
把外层循环放在前面,内层循环依次跟在其后,顺序与原始嵌套结构一致。
1. 普通嵌套循环转为单行列表推导
例如,有如下双层循环:
result = []
for i in range(3):
for j in range(2):
result.append((i, j))
可以写成单行:
result = [(i, j) for i in range(3) for j in range(2)]
输出结果相同:[(0, 0), (0, 1), (1, 0), (1, 1), (2, 0), (2, 1)]
立即学习“Python免费学习笔记(深入)”;
2. 带条件的嵌套循环
如果原循环包含条件判断:
result = []
for i in range(3):
for j in range(3):
if i != j:
result.append((i, j))
对应单行写法:
result = [(i, j) for i in range(3) for j in range(3) if i != j]
3. 多重嵌套也适用
三层循环:
result = []
for a in [1, 2]:
for b in [3, 4]:
for c in [5, 6]:
result.append(a * b * c)
可简化为:
result = [a * b * c for a in [1,2] for b in [3,4] for c in [5,6]]
4. 注意事项
- 虽然写成一行更短,但过度嵌套会影响可读性,建议两层以内使用
- 仅适用于生成列表或类似结构,若需执行复杂逻辑仍建议保留传统循环
- 生成器表达式可用圆括号替代方括号,节省内存:
(x for x in range(10))
基本上就这些。用好列表推导能让代码更 Pythonic,但别为了“一行”牺牲清晰度。











