如何简化嵌套if-elif的Python函数？不拆分多函数的优化方案

核心思路：用字典映射替代多层if-elif #

把不同选项对应的逻辑（包括嵌套的二次选择）用字典结构存储，直接通过输入的字符串匹配对应的处理逻辑，避免嵌套的条件判断，代码更简洁易维护。

输入验证的替代方案（不用try/except） #

可以通过以下方式验证输入合法性：

• 用字符串方法isdigit()判断输入是否为数字字符
• 检查输入是否在预定义的允许选项集合中

完整优化代码示例 #

def func():
    # 定义选项映射：顶层选项 -> 结果/子选项映射
    choice_map = {
        '1': 1,
        '2': {
            '1': 'another result',
            '2': 'random another result'
        },
        '3': {
            '1': 'some result',
            '2': 'some some result',
            '3': 'result'
        }
    }

    while True:
        # 处理顶层输入
        my_choose = input().strip()
        if my_choose not in choice_map:
            print("无效输入，请重新选择")
            continue
        
        current = choice_map[my_choose]
        # 如果是直接结果，返回
        if not isinstance(current, dict):
            return current
        # 如果是子选项，处理二次输入
        while True:
            sub_choose = input().strip()
            if sub_choose in current:
                return current[sub_choose]
            print("无效输入，请重新选择")

代码说明 #

1. 字典映射结构：顶层字典choice_map直接对应原函数的第一层if-elif，需要二次选择的选项对应子字典，和原逻辑完全匹配
2. 输入验证：通过检查输入是否在字典的键集合中，替代try/except处理非法输入，逻辑更直观
3. 循环重试：用while循环处理输入错误，直到用户输入合法选项，避免递归调用（原示例的递归可能导致栈溢出风险）

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