如何根据字符串变量值执行对应函数，优化Python中choice_function避免大量if else

针对你需要替换多分支if-else实现字符串匹配调用函数的需求，最简便易维护的方案是使用字典做函数映射，优化思路和代码如下：

优化逻辑 #

把字符串名称作为字典的key，对应的函数对象作为value，调用时直接通过key查找对应函数执行即可，无需逐个分支判断。

优化后完整代码示例 #

def function1():
    print('function1')
    return


def function2():
    print('function2')
    return


# 集中维护函数名和函数对象的映射关系，新增函数只需在此添加键值对
func_mapping = {
    "function1": function1,
    "function2": function2
}

def choice_function(name):
    # 支持容错处理，匹配不到对应函数时可自定义兜底逻辑
    target_func = func_mapping.get(name)
    if target_func:
        target_func()
    else:
        print(f"无效的函数名：{name}")
    return


def main():
    function_name = 'function1'
    choice_function(function_name)
    function_name = 'function2'
    choice_function(function_name)
    return


if __name__ == '__main__':
    main()

方案优势 #

• 代码简洁：不管有多少个待匹配的函数，都不会出现冗长的if-else链
• 效率更高：字典查找的时间复杂度是O(1)，函数数量越多对比if-else的性能优势越明显
• 易维护：映射关系集中管理，新增/删除函数映射只需要修改字典即可，不需要改动choice_function的逻辑

拓展说明 #

如果待调用的函数需要传参，只需要修改choice_function的调用逻辑即可适配，示例如下：

def choice_function(name, *args, **kwargs):
    target_func = func_mapping.get(name)
    if target_func:
        target_func(*args, **kwargs)
    else:
        print(f"无效的函数名：{name}")

注意：不推荐直接使用globals()[name]()的方式调用，该方法虽然也能实现字符串转函数调用，但会暴露所有全局函数，存在注入执行风险，手动维护映射字典的安全性更高。

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