desarrollar un programa que sume o reste en vectores segun la eleccion del usuario y el numero que ingrese

def add(a, b):
  return a + b
def sub(a, b):
  return a - b
def sub(a, b):
  return a * b
def divi(a, b):
  return a / b
    
add(1, 2)
sub(2,1)
mul(2,1)
div(2,1)