答案是使用列表推导式可将嵌套for循环简化为单行,如[(i, j) for i in range(3) for j in range(2)],支持条件过滤和多层嵌套,但应避免过度嵌套以保持可读性。
在 Python 中,将嵌套 for 循环写成单行通常可以通过列表推导式(List Comprehension)实现。这种方式不仅简洁,而且在多数情况下性能更好。
基本思路:
把外层循环放在前面,内层循环依次跟在其后,顺序与原始嵌套结构一致。
例如,有如下双层循环:
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)]
如果原循环包含条件判断:
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]
三层循环:
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]]
(x for x in range(10))
基本上就这些。用好列表推导能让代码更 Pythonic,但别为了“一行”牺牲清晰度。