crear un menu que solicite elegir

def menu():
  print('1) Suma')
  print('2) Resta')
  print('3) Multiplicacion')
  print('4) Division')
  print('5) Exit')
  option = int(input('Ingrese opcion: '))
  return option

option = menu()