python开发如何将嵌套 for 循环写成单行?

答案是使用列表推导式可将嵌套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)]

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,但别为了“一行”牺牲清晰度。