Python编写简易计算器如何实现非法运算符即时报错

现存问题根因 #

• 运算符校验逻辑完全错误：原代码写的if lark != ops是拿单个输入字符串和整个运算符列表做相等判断，二者类型、内容永远不可能相等，这个判断逻辑完全不符合预期，既没法正确识别合法运算符，也做不到输入非法内容立刻退出。
• 多分支判断写法冗余：连续的if/elif匹配运算逻辑的写法可维护性差，新增运算需要重复加分支，执行效率也有优化空间。
• 额外隐藏bug：运算时两个输入值的顺序写反了，减法、除法场景下计算结果不符合用户输入顺序的预期。

修复&优化思路 #

• 运算符校验改用成员判断：用户输入运算符后立刻用not in判断输入内容是否在合法运算符列表中，不满足就立刻打印提示并退出，不需要等后续输入。
• 用字典映射替代多分支判断：将运算符作为键，对应的双参数计算函数作为值存在字典里，匹配到运算符后直接取对应函数执行即可，不需要逐行分支判断。
• 补充基础的异常兼容：处理数字输入格式错误、除零等常见异常，避免脚本直接崩溃。

优化后完整代码 #

# 简易计算器脚本
error_msg = "非法数学运算符，请仅输入 +、-、*、/ 其中之一"
# 用字典映射运算符和对应计算逻辑，替代多分支判断
op_map = {
    "+": lambda a, b: a + b,
    "-": lambda a, b: a - b,
    "*": lambda a, b: a * b,
    "/": lambda a, b: a / b
}

# 输入运算符后立刻做合法性校验，不合法直接退出
op = input("请输入数学运算符 (+, -, /, *): ")
if op not in op_map:
    print(error_msg)
    quit()

# 获取两个运算值，做基础格式校验
try:
    num1 = float(input("请输入第一个运算值: "))
    num2 = float(input("请输入第二个运算值: "))
except ValueError:
    print("输入错误，请输入有效数字")
    quit()

# 处理除零异常
if op == "/" and num2 == 0:
    print("错误：除数不能为0")
    quit()

# 直接从字典取对应计算逻辑执行
result = op_map[op](num1, num2)
print(f"计算结果为: {result}")

优化点说明 #

• 校验时机前置：运算符输入完成立刻做合法性判断，非法输入直接终止程序，实现即时错误提示的效果。
• 字典映射的写法扩展性更强，后续要新增取余、幂运算等逻辑，只需要在op_map里加新的键值对即可，不需要新增分支判断，代码执行效率更高。
• 用float替代int做类型转换，支持小数运算，适用场景更广。
• 补充了常见异常的处理，脚本运行稳定性更高。

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