Código en Python para ver por consola y guardar en jpg la función seno

print('Autor:' "\033[4;31m"+ 'J Magallanes'+"\033[0;m")

print('Fecha:' "\033[4;31m"+ '28/05/2024'+"\033[0;m")

print('Asignatura:' "\033[4;31m"+ 'Antenas y Propagación'+"\033[0;m")

print('Semestre:' "\033[4;31m"+ '1-2024'+"\033[0;m")

print("Indique 1 para ver las instrucciones, cualquier otro número")

print("para continuar con el programa")

ver=int(input())

if (ver==1):

  print("INTRODUCCIÓN")

  print("------------------------------------------------------------")

  

  print("El programa es capaz de mostrar los resultados por separado")

  print("pero va a generar un solo archivo (resultados.jpg).")

  print("------------------------------------------------------------")

  print("Los datos a ingresar seran los necesarios para lograr generar")

  print("dos vectores de radiacción uno en la dirección Theta y el otro en Phi (en magnitud y fase)")

  print("(si los hay)")

  print("------------------------------------------------------------")

  print("La magnitud es una función mátematica que puede necesitar de los")

  print("siguiente valores:")

  print("1: La corriente")

  print("2:El valor de Theta y Phi para la función trigonometrica (si la hay)")

  print("3: El número de onda")

  print("4: La longitud de la antena")

  print("------------------------------------------------------------")

  print("Ejemplo de la estructura: N1(theta, phi)=I1 sen(k*Longitud*Theta)")

  print("------------------------------------------------------------") 

  print("ó N1(theta, phi)=I1 cos(k*Longitud*Theta) (para este caso se le resta 90 grados)")

  print("Es valida esta estructura tanto para la dirección phi como para la")

  print(" dirección theta.")

else:

  

  print("------------------------------------------------")

  print("elegir una opcción")

  print('1 =  Vector de polarización')

  print('2 =  Razón axial')

  print('3 =  Ángulo de inclinación (tau)')

  print('4 =  Tipo de polarización')

  print('5 =  Gráfica 2D')

  print('6 =  PLF en dB')

  print('7 =  Guardar los resultados obtenidos')

  fg=int(input('Ingrese el número'))

  if (fg==1):

    print('Vector de polarización')

  elif (fg==2):

    print('Razón axial')

  elif (fg==3):

    print('Ángulo de inclinación (tau)')

  elif (fg==4):

    print('Tipo de polarización')

  elif (fg==5):

    print('Gráfica 2D')

    import numpy as np

    import matplotlib.pyplot as plt

    x=np.linspace(0,10,100)

    y=np.sin(x)

    plt.plot(x,y, label='y=sin(x)')

    plt.xlabel('x')

    plt.ylabel('y')

    plt.title('Grafica')

    plt.legend()

    plt.xticks(np.arange(0,10+1,1))

    plt.yticks(np.arange(-1,1+1,1))

    plt.grid(True)

    plt.show()

  elif (fg==6):

    print('PLF en dB')

  else:

    print('Guardar los resultados obtenidos')

    import numpy as np

    import matplotlib.pyplot as plt

    x=np.linspace(0,10,100)

    y=np.sin(x)

    plt.plot(x,y, label='y=sin(x)')

    plt.xlabel('x')

    plt.ylabel('y')

    plt.title('Grafica')

    plt.legend()

    plt.xticks(np.arange(0,10+1,1))

    plt.yticks(np.arange(-1,1+1,1))

    plt.grid(True)

    plt.savefig('Figure.jpg')