Fit a gaussian function

by

Take a look at this answer for fitting arbitrary curves to data. Basically you can use scipy.optimize.curve_fit to fit any function you want to your data. The code below shows how you can fit a Gaussian to some random data (credit to this SciPy-User mailing list post).

import numpy
from scipy.optimize import curve_fit
import matplotlib.pyplot as plt

# Define some test data which is close to Gaussian
data = numpy.random.normal(size=10000)

hist, bin_edges = numpy.histogram(data, density=True)
bin_centres = (bin_edges[:-1] + bin_edges[1:])/2

# Define model function to be used to fit to the data above:
def gauss(x, *p):
    A, mu, sigma = p
    return A*numpy.exp(-(x-mu)**2/(2.*sigma**2))

# p0 is the initial guess for the fitting coefficients (A, mu and sigma above)
p0 = [1., 0., 1.]

coeff, var_matrix = curve_fit(gauss, bin_centres, hist, p0=p0)

# Get the fitted curve
hist_fit = gauss(bin_centres, *coeff)

plt.plot(bin_centres, hist, label="Test data")
plt.plot(bin_centres, hist_fit, label="Fitted data")

# Finally, lets get the fitting parameters, i.e. the mean and standard deviation:
print 'Fitted mean = ', coeff[1]
print 'Fitted standard deviation = ', coeff[2]

plt.show()

More Related Contents:

  1. Histogram Matplotlib
  2. scipy.curve_fit() returns multiple lines
  3. How to draw a histogram from existing bin values
  4. Matplotlib: Specify format of floats for tick labels
  5. A tool to convert MATLAB code to Python [closed]
  6. python numpy/scipy curve fitting
  7. How do I fit a sine curve to my data with pylab and numpy?
  8. Plot two histograms on single chart with matplotlib
  9. Plot a histogram such that bar heights sum to 1 (probability)
  10. matplotlib – extracting data from contour lines
  11. What do all the distributions available in scipy.stats look like?
  12. Fitting a Normal distribution to 1D data
  13. Python curve_fit with multiple independent variables
  14. Linear regression with matplotlib / numpy
  15. How do I put a constraint on SciPy curve fit?
  16. How to plot line (polygonal chain) with numpy/scipy/matplotlib with minimal smoothing
  17. Creating a Confidence Ellipses in a sccatterplot using matplotlib
  18. Logarithmic y-axis bins in python
  19. plotting results of hierarchical clustering on top of a matrix of data
  20. Fitting a histogram with python
  21. Creating a Confidence Ellipsis in a scatterplot using matplotlib
  22. How do I plot list of tuples in Python?
  23. How to center labels in histogram plot
  24. create heatmap2d from txt file
  25. How to have logarithmic bins in a Python histogram
  26. Histogram in matplotlib, time on x-Axis
  27. Interpolating a closed curve using scipy
  28. How to add a mean and median line to a Seaborn displot
  29. combining a log and linear scale in matplotlib
  30. How to display a 3D plot of a 3D array isosurface with mplot3D or similar

Leave a Comment