如何简化Python中含大量条件判断的函数，实现更Pythonic且高性能的写法？

首选方案：字典映射调度表 #

Python中函数是一等公民，可以直接将字符串和对应函数存在字典中，实现O(1)复杂度的查找调度，是这类场景的最优解：

# 提前建立函数名到实际函数的映射表
func_map = {
    'function_1': function_1,
    'function_2': function_2,
    'function_3': function_3,
    'function_4': function_4,
    'function_5': function_5,
    # 新增函数仅需在此处添加键值对即可
}

def apply_function(column, function):
    try:
        return func_map[function](column)
    except KeyError:
        raise ValueError(f"不支持的函数名: {function}") from None

优势： #

• 性能优异：字典查找时间复杂度为O(1)，无论添加多少个函数映射，查询速度都不会下降，相比if-elif链的*O(n)*复杂度，分支越多性能提升越明显
• 可维护性强：新增/删除/修改映射关系仅需修改func_map，无需改动apply_function的核心逻辑，符合开闭原则
• 代码简洁：省去大量重复的条件判断代码，可读性更高

其他可选方案 #

方案2：match-case语法（Python 3.10+） #

如果分支数量较少（10个以内），且使用Python 3.10及以上版本，可以用match-case简化if-elif链，可读性比传统if-elif更好，但性能和if-elif基本持平：

def apply_function(column, function):
    match function:
        case 'function_1':
            return function_1(column)
        case 'function_2':
            return function_2(column)
        case 'function_3':
            return function_3(column)
        case _:
            raise ValueError(f"不支持的函数名: {function}")

方案3：动态获取全局函数（不推荐） #

如果函数名和传入的字符串完全一致，也可以通过globals()动态获取函数，但是该方案存在安全风险，仅适合完全可控的内部场景，禁止在function参数为外部用户可输入的场景使用：

def apply_function(column, function):
    try:
        return globals()[function](column)
    except KeyError:
        raise ValueError(f"不支持的函数名: {function}") from None

注意：该方案可能导致恶意注入执行未授权函数，非必要不使用。

选型建议 #

• 绝大多数场景优先选择字典映射方案，兼顾性能、安全性和可维护性，是最符合Python风格的实现
• 分支数量少且Python版本符合要求时可以选择match-case方案
• 严格避免在不可信输入场景使用动态获取函数的方案

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