如何利用用户输入调用对应函数，避免大量if条件判断？

在Python里，处理这种根据不同输入调用对应函数的场景，用字典映射是最优解——它能把O(n)的if-else遍历变成O(1)的字典查找，代码也更整洁易维护。

基础用法示例 #

直接把输入字符串作为键，函数对象作为值存入字典，输入后通过键取值调用：

def a():
    print("You entered A")

def b():
    print("You entered B")

# 构建输入与函数的映射表
func_mapping = {
    "a": a,
    "b": b,
    # 新增函数只需在这里添加键值对，无需修改判断逻辑
}

user_input = input("Input: ")

# 调用对应函数，同时处理无效输入
if user_input in func_mapping:
    func_mapping[user_input]()
else:
    print(f"无效输入: {user_input}")

进阶：带参数的函数调用 #

如果函数需要接收参数，只需在调用时传递对应参数即可：

def add(x, y):
    print(f"结果: {int(x) + int(y)}")

def multiply(x, y):
    print(f"结果: {int(x) * int(y)}")

func_mapping = {
    "add": add,
    "multiply": multiply,
}

# 示例输入：add 3 5
user_input = input("输入命令和参数: ").split()
if user_input[0] in func_mapping:
    # 拆分命令和参数，用*解包传递
    func_mapping[user_input[0]](*user_input[1:])
else:
    print("无效命令")

更简洁的无效输入处理 #

用dict.get()方法可以一行搞定调用+默认处理：

func_mapping.get(user_input, lambda: print("无效输入"))()

为什么字典映射更好？ #

• 效率高：字典的键查找是O(1)时间复杂度，上百个函数时比遍历if-else快得多
• 易维护：新增/删除函数只需修改映射字典，不用动判断逻辑
• 扩展性强：支持带参函数、类方法、lambda表达式等多种场景

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