Ejemplo n.º 1
0
import CurveFit as cf
import matplotlib.pyplot as plt

# QUESTAO 1

# a) plota a dispersao dos dados-------------------------------------------------
xi = np.array([1872, 1890, 1900, 1920, 1940, 1950, 1960, 1970, 1980, 1991, 1996])

yi = np.array([9.9, 14.3, 17.4, 30.6, 41.2, 51.9, 70.2, 93.1, 119.0, 146.2, 157.1])

plt.plot(xi, yi, 'ro')
# ---------------------------------------------------------------------------------


# b) calcula a funcao quadrática e os resultados------------------------------------
resultsp, polynomial = cf.fitpolynomial(xi, yi, 2)

# Cria a linha da funcao e plota
xx = np.linspace(xi[0], xi[-1])
plt.plot(xx, np.polyval(polynomial,xx),'b-')
# -----------------------------------------------------------------------------------


# c) calcula a funcao exponencial(A e b) e os resultados ---------- -----------------
resultse, A, b = cf.fitexponencial(xi, yi)

# Cria a linha da funcao e plota usando a Ae^(b*x)
xx = np.linspace(xi[0], xi[-1])  
plt.plot(xx, A * np.exp(b*xx),'g-')

plt.grid();plt.show()