如何简化Python嵌套循环中大量的list append操作？

先假设你的原始代码大概是这样（模拟嵌套append的场景）：

import random

def random_num():
    return random.randint(1, 100)

result = []
for i in range(5):
    for j in range(3):
        # 实际场景中这里可能是带大量索引的feature函数调用
        val = random_num()
        result.append(val)

针对这类代码，有几种直接的简化方式：

1. 使用列表推导式 #

列表推导式天生适合替代这种嵌套循环+append的结构，代码更紧凑且可读性更强：

import random

def random_num():
    return random.randint(1, 100)

result = [random_num() for i in range(5) for j in range(3)]

如果feature函数需要用到循环变量（比如i、j），直接在推导式里传入即可：

# 假设feature需要i和j作为参数
def feature(i, j):
    return i * 10 + j

result = [feature(i, j) for i in range(5) for j in range(3)]

2. 利用itertools.product处理笛卡尔积场景 #

如果嵌套循环是遍历多个可迭代对象的笛卡尔积，用itertools.product可以避免手动写多层循环，尤其适合3层及以上的嵌套：

import random
from itertools import product

def random_num():
    return random.randint(1, 100)

# product(range(5), range(3)) 等价于两层嵌套循环的所有(i,j)组合
result = [random_num() for _ in product(range(5), range(3))]
# 如果需要用到循环变量
result = [feature(i, j) for i, j in product(range(5), range(3))]

3. 生成器表达式（按需生成） #

如果不需要立即生成完整列表，只是后续迭代使用，用生成器表达式更节省内存，语法和列表推导式类似，把方括号换成圆括号即可：

gen = (random_num() for i in range(5) for j in range(3))
# 后续需要时再转成列表或直接迭代
result = list(gen)

4. 优化feature函数的索引操作 #

如果feature函数里有大量索引操作导致代码混乱，可以：

• 用解包替代索引：比如把arr[0], arr[1], arr[2]改成a, b, c = arr
• 用zip批量处理多组索引对应的元素：比如同时遍历多个列表的对应位置元素时，用for x, y, z in zip(list1, list2, list3)替代手动索引
• 把重复的索引逻辑封装成小函数，避免在循环里重复写复杂的索引计算

内容的提问来源于stack exchange，提问作者Megan