如何以Pythonic方式优化嵌套循环中列表调用顺序的重复代码

核心思路 #

1. 用字典统一管理所有属性及其可选值，避免分散定义导致的重复维护。
2. 固定命名格式的属性顺序，确保最终生成的图表名称严格遵循{color}_{shape}_{size}_{number}格式。
3. 利用itertools.product处理任意顺序、任意长度的循环属性组合，替代手写嵌套循环消除重复代码。
4. 通过属性索引映射或字典更新，将循环生成的属性组合映射到固定命名顺序上。

完整代码示例 #

场景1：所有属性均参与循环，仅调整循环顺序

import itertools

# 统一管理属性与可选值
attr_options = {
    'color': ['red', 'blue', 'green'],
    'shape': ['circle', 'square', 'triangle'],
    'size': ['small', 'medium', 'large'],
    'number': ['1', '2', '3']
}

# 固定的命名格式顺序
NAME_ORDER = ['color', 'shape', 'size', 'number']

def generate_charts(loop_by):
    # 校验输入的循环属性合法性
    if not all(attr in attr_options for attr in loop_by):
        raise ValueError("loop_by包含无效属性")
    
    # 按循环顺序提取对应属性的取值列表
    loop_values = [attr_options[attr] for attr in loop_by]
    # 建立属性到循环组合中索引的映射
    attr_index_map = {attr: loop_by.index(attr) for attr in NAME_ORDER}
    
    # 生成所有循环组合
    for combo in itertools.product(*loop_values):
        # 按命名顺序提取对应值并拼接
        name_parts = [combo[attr_index_map[attr]] for attr in NAME_ORDER]
        chart_name = '_'.join(name_parts)
        
        # 替换为实际生成图表的逻辑
        print(f"生成图表: {chart_name}")

# 示例调用：按size -> color的顺序循环
generate_charts(loop_by=['size', 'color'])
# 示例调用：按shape -> number -> color -> size的顺序循环
generate_charts(loop_by=['shape', 'number', 'color', 'size'])

场景2：部分属性固定，仅循环指定属性

如果需要固定某些属性值，只循环部分属性，可以扩展函数支持传入固定参数：

import itertools

attr_options = {
    'color': ['red', 'blue', 'green'],
    'shape': ['circle', 'square', 'triangle'],
    'size': ['small', 'medium', 'large'],
    'number': ['1', '2', '3']
}

NAME_ORDER = ['color', 'shape', 'size', 'number']

def generate_charts(loop_by, fixed_attr=None):
    fixed_attr = fixed_attr or {}
    # 合并固定属性与默认可选值
    all_attrs = {**attr_options, **fixed_attr}
    
    if not all(attr in all_attrs for attr in loop_by):
        raise ValueError("loop_by包含无效属性")
    
    loop_values = [all_attrs[attr] for attr in loop_by]
    # 初始化完整属性模板，填充固定值
    full_attr_template = {attr: fixed_attr.get(attr) for attr in NAME_ORDER}
    
    for combo in itertools.product(*loop_values):
        # 更新模板中的循环属性值
        current_attrs = full_attr_template.copy()
        current_attrs.update(dict(zip(loop_by, combo)))
        
        # 按命名顺序拼接名称
        name_parts = [current_attrs[attr] for attr in NAME_ORDER]
        chart_name = '_'.join(name_parts)
        
        print(f"生成图表: {chart_name}")

# 示例调用：固定size为medium、number为2，按color -> shape循环
generate_charts(loop_by=['color', 'shape'], fixed_attr={'size': 'medium', 'number': '2'})

方案优势 #

• 消除重复代码：无需为不同的循环顺序手写多层嵌套循环，一套逻辑适配所有情况。
• 严格遵循命名规则：通过固定的NAME_ORDER确保生成的图表名称格式完全符合要求。
• 高扩展性：新增或修改属性时，只需更新attr_options和NAME_ORDER，无需改动核心循环逻辑。
• Pythonic风格：利用标准库itertools处理笛卡尔积，结合字典推导、列表推导简化映射逻辑，代码简洁易读。

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