Book Image

IPython Interactive Computing and Visualization Cookbook

By : Cyrille Rossant
Book Image

IPython Interactive Computing and Visualization Cookbook

By: Cyrille Rossant

Overview of this book

Table of Contents (22 chapters)
IPython Interactive Computing and Visualization Cookbook
Credits
About the Author
About the Reviewers
www.PacktPub.com
Preface
Index

Fitting a function to data with nonlinear least squares


In this recipe, we will show an application of numerical optimization to nonlinear least squares curve fitting. The goal is to fit a function, depending on several parameters, to data points. In contrast to the linear least squares method, this function does not have to be linear in those parameters.

We will illustrate this method on artificial data.

How to do it…

  1. Let's import the usual libraries:

    In [1]: import numpy as np
            import scipy.optimize as opt
            import matplotlib.pyplot as plt
            %matplotlib inline
  2. We define a logistic function with four parameters:

    In [2]: def f(x, a, b, c, d):
                return a/(1 + np.exp(-c * (x-d))) + b
  3. Let's define four random parameters:

    In [3]: a, c = np.random.exponential(size=2)
            b, d = np.random.randn(2)
  4. Now, we generate random data points by using the sigmoid function and adding a bit of noise:

    In [4]: n = 100
            x = np.linspace(-10., 10., n)
            y_model = f(x, a, b,...