Python优化：如何替换for循环中的if语句以优化代码结构

原代码存在逻辑偏差（循环内错误判断整个sublist而非单个元素）、重复代码冗余、操作步骤繁琐的问题，用字典映射颜色与对应列表可以彻底消除重复，同时简化流程：

# 用字典统一管理颜色和对应的目标列表
color_mapping = {
    "red": [],
    "blue": [],
    "green": []
}

# 先确认当前sublist里存在哪些目标颜色
existing_colors = [color for color in color_mapping if color in sublist]

# 一次遍历完成过滤和追加操作
for item in sublist:
    # 跳过None值和颜色本身
    if item is None or item in color_mapping:
        continue
    # 给所有存在的颜色对应的列表添加元素
    for color in existing_colors:
        color_mapping[color].append(item)

# 如需单独的列表变量，直接从字典提取即可
red_list = color_mapping["red"]
blue_list = color_mapping["blue"]
green_list = color_mapping["green"]

核心优化点 #

• 用字典替代三个独立列表，避免重复定义和重复操作
• 提前筛选出当前sublist中存在的颜色，减少循环内的判断次数
• 合并过滤（移除None和颜色字符串）与追加逻辑，省去原代码中filter、remove等冗余步骤
• 避免原代码中remove可能引发的「元素不存在」报错问题

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