Curve fittingΒΆ

A curve fitting example

import numpy as np
from scipy import optimize
import pylab as pl
# Our test function
def f(t, omega, phi):
return np.cos(omega * t + phi)
# Our x and y data
x = np.linspace(0, 3, 50)
y = f(x, 1.5, 1) + .1*np.random.normal(size=50)
# Fit the model: the parameters omega and phi can be found in the
# `params` vector
params, params_cov = optimize.curve_fit(f, x, y)
# plot the data and the fitted curve
t = np.linspace(0, 3, 1000)
pl.plot(x, y, 'bx')
pl.plot(t, f(t, *params), 'r-')

Total running time of the script: ( 0 minutes 0.038 seconds)

Generated by Sphinx-Gallery