Python基础作业题编写numu函数

def menu():
    a = input('1)加法:\n2)减法：\n请输入1或2：')
    if a == '1' or a == '2':
        print("输入正确")
    else:
        print("输入错误，请输入‘1’或‘2’")