编辑代码

import math

def ZBW_add(a, b):
    return a + b
#定义加法函数
def ZBW_subtract(a, b):
    return a - b
#定义减法函数
def ZBW_multiply(a, b):
    return a * b
#定义乘法函数
def ZBW_division(a, b):
    try:
        return a / b
    except ZeroDivisionError:
        print("除数不能为零")
        return None
#定义除法函数以及异常处理
def ZBW_squareroot(a):
    if a < 0:
        print("不能计算负数的平方根")
        return None
    return math.sqrt(a)
#定义平方根以及处理负数的情况
def ZBW_power(a, b):
    return a ** b
#定义一个幂运算函数，返回a的b次方
def main():
    while True:
        print("请选择你的操作:")
        print("1. ZBW_add")
        print("2. ZBW_subtract")
        print("3. ZBW_multiply")
        print("4. ZBW_division")
        print("5. ZBW_squareroot")
        print("6. ZBW_power")
        print("7. 退出")
        try:
            choice = int(input('请输入选择的数:'))
            if choice == 7:
                print("程序退出")
                break
            elif choice < 1 or choice > 7:
                print("无效的选择，请重新输入")
                continue
            if choice != 5:
                a = float(input('请输入第一个数:'))
                b = float(input('请输入第二个数:'))
            else:
                a = float(input('请输入一个数:'))
#根据用户选择调用相应的函数
            if choice == 1:
                result = ZBW_add(a, b)
            elif choice == 2:
                result = ZBW_subtract(a, b)
            elif choice == 3:
                result = ZBW_multiply(a, b)
            elif choice == 4:
                result = ZBW_division(a, b)
            elif choice == 5:
                result = ZBW_squareroot(a)
            elif choice == 6:
                result = ZBW_power(a, b)
            if result is not None:
                print(f"结果是: {result}")
        except ValueError:
            print("请输入有效的数字或选择")


if __name__ == "__main__":
    main()