如何用Python将7层嵌套for循环重构为更易扩展的优雅实现？

核心思路 #

用itertools.product生成所有参数的笛卡尔积替代多层嵌套循环，同时把字段名与对应参数列表做成可配置的映射结构，后续新增参数维度（比如湿度、第二个衰减器）只需要在配置里加一行，完全不用修改循环逻辑，扩展性拉满。

优化后代码 #

import itertools

def create_ref_dict(self):
    # 配置：字段名 -> 对应参数列表，新增维度直接加在这里
    param_config = [
        ('Oven_Temperature', self.temperature_list),
        ('Tag_Angle', self.angle_list),
        ('User_GW_DCA_Raw', self.att_list),
        ('GW_Timing_List', self.timing_profile_list),
        ('Energizing_Pattern_List', self.ep_list),
        ('GW_Beacon_BO', self.beacon_bo_list),
        ('Number_of_Test_Iterations', self.number_of_iterations)
    ]
    
    # 拆分字段名和参数列表，用于生成笛卡尔积和构建字典
    field_names, param_lists = zip(*param_config)
    
    idx = 0
    # 生成所有参数组合，替代多层嵌套循环
    for params in itertools.product(*param_lists):
        if idx >= self.total_size:
            break
        # 用字段名和对应参数构建字典
        tmp_dict = dict(zip(field_names, params))
        self.ref_dict[idx] = tmp_dict
        idx += 1

为什么这么改？ #

• 扩展性强：要加湿度参数？只需要在param_config里加一行('Humidity', self.humidity_list)；要加第二个衰减器？再加一行('Second_Attenuator', self.att2_list)，循环逻辑完全不动。
• 可读性高：一眼就能看到所有参与组合的参数和对应字段，比7层嵌套直观太多。
• 代码简洁：把重复的嵌套逻辑交给itertools.product处理，不用写一堆嵌套for循环。

额外说明 #

原代码里的while idx < self.total_size逻辑保留了，避免生成的组合总数超过total_size时出现越界。如果能确保total_size等于所有参数列表长度的乘积，这个判断也可以去掉，直接遍历所有组合即可。

内容的提问来源于stack exchange，提问作者בר חיימוביץ