如何实现支持自定义运算数数量的Python可扩展计算器？

核心思路 #

不用硬编码数字数量或写大量if判断，而是通过循环收集用户输入的数字，再用累积运算的方式处理任意数量的数字，同时修复原代码里的退出逻辑问题。

优化后的完整代码 #

import operator

# 定义运算映射
ops = {
    "*": operator.mul,
    "/": operator.truediv,
    "+": operator.add,
    "-": operator.sub,
    "^": operator.pow
}

def get_numbers():
    """获取用户指定数量的数字，返回数字列表"""
    while True:
        try:
            num_count = int(input("请输入参与运算的数字数量（至少2个）："))
            if num_count >= 2:
                break
            print("错误：数字数量必须大于等于2！")
        except ValueError:
            print("无效输入，请输入整数！")
    
    numbers = []
    for i in range(num_count):
        while True:
            try:
                num = float(input(f"请输入第{i+1}个数字："))
                numbers.append(num)
                break
            except ValueError:
                print("无效输入，请输入数字！")
    return numbers

# 启动提示
print("欢迎使用计算器！\n")
print("可选运算：")
print("\n".join(ops.keys()))

calc_stop = False
while not calc_stop:
    # 选择运算
    choice = input("\n请选择运算：").strip()
    if choice not in ops:
        print("无效运算，请重新选择！")
        continue
    
    # 获取数字列表
    numbers = get_numbers()
    
    # 执行累积运算
    try:
        result = numbers[0]
        for num in numbers[1:]:
            result = ops[choice](result, num)
    except ZeroDivisionError:
        print("错误：除数不能为0！")
        continue
    
    # 输出结果
    equation = f" {' '.join([choice, str(num)])}".join(map(str, numbers))
    print(f"{equation} = {result}")
    
    # 询问是否继续
    while True:
        next_calc = input("是否继续计算？(y/n) ").strip().lower()
        if next_calc == 'n':
            print("再见！")
            calc_stop = True
            break
        elif next_calc == 'y':
            print("准备下一次计算...\n")
            break
        else:
            print("无效输入，请输入y或n！")

关键优化点 #

1. 动态收集数字：

   • 用get_numbers函数循环获取用户指定数量的数字，不管用户输入2个、3个还是更多，都能统一处理，彻底避免了大量if判断。
   • 加入输入验证，确保用户输入的是有效数字和符合要求的数量。

2. 累积运算逻辑：

   • 从第一个数字开始，依次和后续数字执行选择的运算，最终得到结果。
   • 新增了除零错误的捕获，提升程序健壮性。

3. 简化退出逻辑：

   • 去掉原代码中冗余的for循环，直接判断用户输入的y/n，逻辑更清晰。

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