Python菜单程序中重复if条件与变量定义的优化方法咨询

嗨，Rafael！作为Python新手遇到这种重复代码的问题太正常了——我刚学的时候也被一堆重复的if和变量定义搞得头大😅。你现在的核心问题是违反了DRY（Don't Repeat Yourself）原则，也就是把相同的输入逻辑写了N遍，其实我们可以用几种简单的方式来优化，而且完全适合初学者上手：

方法1：把重复的输入逻辑封装成函数 #

最直接的方式就是把那些重复的input语句抽出来做成一个函数，不管哪个菜单选项需要这些变量，直接调用这个函数就行，不用每次都重新写一遍。

示例代码：

def get_user_inputs():
    # 把重复的输入逻辑放在这里
    a = int(input('输入A的值: '))
    b = int(input('输入B的值: '))
    c = int(input('输入C的值: '))
    d = int(input('输入D的值: '))
    return a, b, c, d

# 你的菜单主逻辑可以简化成这样
x = int(input("请选择菜单选项（1/2/3）: "))
if x == 1:
    a, b, c, d = get_user_inputs()
    # 这里写选项1独有的处理逻辑，比如计算、打印等
elif x == 2:
    a, b, c, d = get_user_inputs()
    # 选项2独有的处理逻辑
elif x == 3:
    a, b, c, d = get_user_inputs()
    # 选项3独有的处理逻辑
else:
    print("无效选项，请重新选择！")

这样做的好处很明显：以后如果要修改输入提示（比如把“输入A的值”改成“请输入第一个整数”），或者要加输入验证（比如判断用户输入的是不是整数），只需要修改get_user_inputs()这一个函数就行，不用每个if分支都改一遍，代码也清爽多了。

方法2：用字典映射替代一堆if-elif（进阶一点，但同样好上手） #

如果每个菜单选项除了输入，还有不同的处理逻辑，我们可以把每个选项的处理逻辑也封装成函数，然后用一个字典把“选项数字”和“对应的处理函数”关联起来，这样连if-elif都可以省掉，代码结构更清晰。

示例代码：

def get_user_inputs():
    a = int(input('输入A的值: '))
    b = int(input('输入B的值: '))
    c = int(input('输入C的值: '))
    d = int(input('输入D的值: '))
    return a, b, c, d

# 每个选项的处理函数
def handle_option1():
    a, b, c, d = get_user_inputs()
    print(f"正在执行选项1：A={a}, B={b}, C={c}, D={d}")
    # 这里写选项1的具体业务逻辑

def handle_option2():
    a, b, c, d = get_user_inputs()
    print(f"正在执行选项2：A={a}, B={b}, C={c}, D={d}")
    # 选项2的具体业务逻辑

def handle_option3():
    a, b, c, d = get_user_inputs()
    print(f"正在执行选项3：A={a}, B={b}, C={c}, D={d}")
    # 选项3的具体业务逻辑

# 菜单映射：键是选项数字，值是对应的处理函数
menu_map = {
    1: handle_option1,
    2: handle_option2,
    3: handle_option3
}

# 主逻辑
x = int(input("请选择菜单选项（1/2/3）: "))
if x in menu_map:
    menu_map[x]()  # 调用对应的处理函数
else:
    print("无效选项，请重新选择！")

这种方式的优势在于扩展性极强——以后要加新选项，只需要写一个新的处理函数，然后在menu_map里加一行键值对就行，完全不用动主逻辑的代码。

关于类的小建议 #

你提到尝试过类但不太顺手，其实类是更高级的封装方式，但对于初学者来说，先把函数和字典映射用熟练，再慢慢过渡到类会更轻松。比如可以把整个菜单程序封装成一个MenuProgram类，里面包含输入方法、处理方法和运行方法，但这一步可以等你对函数更熟悉之后再尝试。

刚开始学编程不用着急，慢慢体会“代码复用”的思路，多写几次就会越来越顺手啦！

备注：内容来源于stack exchange，提问作者Rafael Silva