Python如何将多层if/else函数调用改为字典映射并处理非法键？

现有写法的问题 #

你设想的字典映射是非常典型的Pythonic多分支替代方案，比堆叠if-else的可维护性强得多，但你给出的示例代码存在逻辑问题：字典初始化阶段就会执行所有func1/func2/func3，无论你传入的a值是什么，不仅浪费性能，也不符合按需调用的预期。

正确实现方案 #

字典的value存函数对象本身，而非函数执行结果，拿到对应函数后再传入参数调用即可，同时可以用字典的get方法处理key不存在的场景：

def choose_function(a, b, c):
    # 字典value存储函数对象，不提前执行
    func_options = {
        'first_option': func1,
        'second_option': func2,
        'third_option': func3
    }
    # 用get方法获取目标函数，key不存在时返回None
    target_func = func_options.get(a)
    if target_func is None:
        # 可根据业务需求选择处理逻辑：抛异常/返回默认值/走兜底函数
        raise ValueError(f"传入的选项 {a} 不在支持范围内")
    # 调用对应函数并返回结果
    return target_func(b, c)

后续要新增分支只需要给func_options添加新的键值对即可，不需要修改逻辑分支，维护成本远低于if-else堆叠的写法。

进阶优化（兼容参数不一致的函数） #

如果后续新增的函数参数数量、类型不统一，可以改用可变参数适配，不用每次修改choose_function的入参定义：

def choose_function(a, *args, **kwargs):
    func_options = {
        'first_option': func1,
        'second_option': func2,
        'third_option': func3
        # 后续新增分支只需要加键值对即可
    }
    target_func = func_options.get(a)
    if not target_func:
        raise ValueError(f"不支持的选项: {a}")
    return target_func(*args, **kwargs)

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