Fit mixture of two gaussian/normal distributions to a histogram from one set of data, python

by

Here a simulation with scipy tools :

from pylab import *
from scipy.optimize import curve_fit

data=concatenate((normal(1,.2,5000),normal(2,.2,2500)))
y,x,_=hist(data,100,alpha=.3,label="data")

x=(x[1:]+x[:-1])/2 # for len(x)==len(y)

def gauss(x,mu,sigma,A):
    return A*exp(-(x-mu)**2/2/sigma**2)

def bimodal(x,mu1,sigma1,A1,mu2,sigma2,A2):
    return gauss(x,mu1,sigma1,A1)+gauss(x,mu2,sigma2,A2)

expected=(1,.2,250,2,.2,125)
params,cov=curve_fit(bimodal,x,y,expected)
sigma=sqrt(diag(cov))
plot(x,bimodal(x,*params),color="red",lw=3,label="model")
legend()
print(params,'\n',sigma)

The data is the superposition of two normal samples, the model a sum of Gaussian curves. we obtain :

gauss with legend

And the estimate parameters are :

# via pandas :
# pd.DataFrame(data={'params':params,'sigma':sigma},index=bimodal.__code__.co_varnames[1:])
            params     sigma
mu1       0.999447  0.002683
sigma1    0.202465  0.002696
A1      226.296279  2.597628
mu2       2.003028  0.005036
sigma2    0.193235  0.005058
A2      117.823706  2.658789

More Related Contents:

  1. How do I read CSV data into a record array in NumPy?
  2. Working with big data in python and numpy, not enough ram, how to save partial results on disc?
  3. Python out of memory on large CSV file (numpy)
  4. python numpy/scipy curve fitting
  5. Multiple linear regression in Python
  6. matplotlib – extracting data from contour lines
  7. Fitting a Normal distribution to 1D data
  8. Populate a Pandas SparseDataFrame from a SciPy Sparse Matrix
  9. Installing NumPy and SciPy on 64-bit Windows (with Pip)
  10. How to extend an array in-place in Numpy?
  11. Numpy Resize/Rescale Image
  12. Interpolation over an irregular grid
  13. fitting exponential decay with no initial guessing
  14. Overflow Error in Python’s numpy.exp function
  15. How to plot line (polygonal chain) with numpy/scipy/matplotlib with minimal smoothing
  16. How to calculate the statistics “t-test” with numpy
  17. How can I efficiently process a numpy array in blocks similar to Matlab’s blkproc (blockproc) function
  18. Creating a Confidence Ellipsis in a scatterplot using matplotlib
  19. How do I plot list of tuples in Python?
  20. generalized cumulative functions in NumPy/SciPy?
  21. Principal component analysis in Python
  22. How to convert a column or row matrix to a diagonal matrix in Python?
  23. How to do a polynomial fit with fixed points
  24. How to smooth a curve for a dataset
  25. How to transform numpy.matrix or array to scipy sparse matrix
  26. Matplotlib – Finance volume overlay
  27. sparse 3d matrix/array in Python?
  28. Convert Pandas dataframe to Sparse Numpy Matrix directly
  29. Structure of inputs to scipy minimize function
  30. ValueError: Dimension mismatch

Leave a Comment