Python如何仅按需计算指定函数结果，避免全量执行？

问题背景 #

想要实现一个根据传入字母参数返回对应结果的函数，不想用冗长的多分支if-elif写法：

def get_result(letter, value):
    if letter == "a":
        return(get_result_a(value))
    elif letter == "b":
        return(get_result_b(value))
    elif letter == "c":
        return(get_result_c(value))
    elif letter == "d":
        return(get_result_d(value))
    elif letter == "e":
        return(get_result_e(value))

改成列表匹配的写法后，发现Python会提前执行列表里所有函数，即便只需要其中一个，参数多的时候性能很差：

def get_result(letter, value):
    results = [get_result_a(value), get_result_b(value), get_result_c(value), get_result_d(value), get_result_e(value)]
    letters = ["a", "b", "c", "d", "e"]
    for i in range(len(results)):
        if letter == letters[i]:
            return results[i]

优化方案 #

用字典映射函数对象，而不是提前执行函数。这样只有当匹配到对应的字母时，才会调用对应的函数并传入参数，避免不必要的计算：

def get_result(letter, value):
    # 构建字母到函数的映射，存储的是函数对象而非执行结果
    func_map = {
        "a": get_result_a,
        "b": get_result_b,
        "c": get_result_c,
        "d": get_result_d,
        "e": get_result_e
    }
    # 获取对应的函数，若字母不存在可设置默认处理（比如返回None或抛出异常）
    target_func = func_map.get(letter)
    if target_func:
        return target_func(value)
    # 处理字母不在映射中的情况
    raise ValueError(f"Unsupported letter: {letter}")

补充说明 #

• 字典的键是匹配的字母，值是未执行的函数对象，不会提前计算结果
• 使用dict.get()可以安全处理未匹配到的情况，避免KeyError
• 如果需要扩展更多字母，只需在func_map中添加新的键值对，比if-elif更易维护

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