import numpy as np

from os import system

from sympy import symbols,init_printing,diff,integrate
x,y,z=symbols("x y z")
init_printing()
op="0"
while op!="":
print("""
MENU DE OPÇÕES
[1] EQUAÇÃO DE 1° GRAU
[2] EQUAÇÃO DE 2° GRAU
[3] AJUSTE DE CURVAS
[4] DERIVADA DE UMA FUNÇÃO
[5] INTEGRAL
""")
op=input("digite um dos números das opções acima[ENTER PARA FECHAR]: ")
print()
while op not in ["","1","2","3","4","5"]:
op=input("opção invalida tente novamente: ")
if op=="1":
print("digite os coeficientes:")
a=float(input("a: "))
b=float(input("b: "))
P=np.poly1d([a,b])
print(P)
print()
subs="S"
while subs=="S":
x=float(input("digite um valor que deseje substituir: "))
print("P(%.2f)=%.2f"%(x,P(x)))
subs=input("digite S para continuar e N para fechar: ").upper()
system("clear")
while subs not in ["S","N"]:
subs=input("digite S para continuar e N para fechar: ").upper()
if op=="2":
print("digite os coeficientes:")
a=float(input("a: "))
b=float(input("b: "))
c=float(input("c: "))
P=np.poly1d([a,b,c])
print(P)
print()
subs="S"
while subs=="S":
x=float(input("digite um valor que deseje substituir: "))
print("P(%.2f)=%.2f"%(x,P(x)))
subs=input("digite S para continuar e N para fechar: ").upper()
system("clear")
while subs not in ["S","N"]:
 subs=input("digite S para continuar e N para fechar: ")
if op=="3":
pontos=int(input("digite o número de pontos almejados: "))
xl=[]
yl=[]
i=1
for c in range(pontos):
x=float(input("digite o valor de x do ponto %i: "%i))
y=float(input("digite o valor de y do ponto %i: "%i))
xl.append(x)
yl.append(y)
i+=1
xl=np.array(xl)
yl=np.array(yl)
grau=float(input("digite o grau a ser ajustado: "))
P=np.poly1d(np.polyfit(xl,yl,grau))
print(P)
print()
if op=="4":
F=input("digite a função que almeje derivar(obs.: multiplicação tem
que colocar o sinal de *: ")
Fd=diff(F,x)
print()
print(Fd)
if op=="5":
F=input("digite a função que almeje integrar(obs.: multiplicação tem
que colocar o sinal de *: ")
FI=integrate(F)
print()
print(FI,"+ C")
if op=="":
print("FIM DE OPERAÇÃO")