Python计算器多数字运算优化：乘除三数运算逻辑改进咨询

嘿，刚学Python就能写出计算器，已经很棒啦！咱们来聊聊你遇到的问题，以及更简洁的解决思路～

问题根源 #

你现在的麻烦在于加减和乘除的“单位元”不一样：

• 加减里，0是单位元——加0、减0都不会改变结果，所以默认设Number3=0没问题；
• 但乘除里，单位元是1——乘1、除以1才不改变结果，默认设0就会直接把乘法结果变成0，除法还会触发除以0的错误（你原来的代码里如果用户没输入第三个数却选了除法，其实会报错ZeroDivisionError，只是你用Number4=1绕开了）。

用额外的Number4确实能解决问题，但代码会变得冗余，咱们可以用更优雅的方式重构。

最优实现思路：用可变参数+数字列表 #

核心思路是先收集所有用户要计算的数字，再根据运算类型统一处理，而不是固定死3个参数。这样不仅解决了默认值问题，还能支持任意多个数字的运算，扩展性更强。

改进后的代码

import math

print("Python calculator")
print("""Select an operation below
1: Add
2: Sub
3: Mult
4: Div
""")

# 定义支持任意多个数字的运算函数
def add(*args):
    return sum(args)

def sub(*args):
    result = args[0]
    for num in args[1:]:
        result -= num
    return result

def mult(*args):
    # Python 3.8+ 可用math.prod，也可以手动实现兼容旧版本
    return math.prod(args)
    # 手动实现版本：
    # result = 1
    # for num in args:
    #     result *= num
    # return result

def div(*args):
    result = args[0]
    for num in args[1:]:
        if num == 0:
            return "Error: Cannot divide by zero!"
        result /= num
    return result

# 收集用户输入的数字
operation = input("Enter Operation Number: ")
numbers = []
numbers.append(int(input("Enter Your First Number: ")))
numbers.append(int(input("Enter your Second Number: ")))

while True:
    more = input("Do you have more numbers? (Yes or No): ").strip().lower()
    if more == "yes":
        numbers.append(int(input("What is your other number? ")))
    elif more == "no":
        break
    else:
        print("Please enter Yes or No!")

# 执行运算并输出结果
print("Calculating...")
if operation == '1':
    print(add(*numbers))
elif operation == '2':
    print(sub(*numbers))
elif operation == '3':
    print(mult(*numbers))
elif operation == '4':
    print(div(*numbers))
else:
    print("Invalid operation number!")

代码优势说明 #

• 摆脱固定参数限制：用*args可变参数，支持用户输入2个、3个甚至更多数字，不用再纠结默认值的问题；
• 逻辑统一简洁：不管是加减还是乘除，都直接用收集到的数字列表处理，不需要额外的分支判断“有没有第三个数”；
• 增加错误处理：除法里加了除以0的判断，避免程序崩溃；
• 更友好的输入循环：用户输入非Yes/No时会提示重新输入，体验更好。

针对你原有代码的轻量调整（如果不想大改） #

如果暂时不想重构整个代码，也可以把乘除函数的第三个参数设默认值为1，替代临时的Number4：

# 修改乘除函数，给第三个参数设默认值1
def mult(x, y, z=1):
    return x * y * z

def div(x, y, z=1):
    if z == 0:
        return "Error: Cannot divide by zero!"
    return x / y / z

# 调用时直接传Number3即可，无需额外分支
if Operation == '3':
    print(mult(Number1, Number2, Number3))
elif Operation == '4':
    print(div(Number1, Number2, Number3))

这样也能避免冗余变量，让逻辑更清爽～

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