How to fit polynomial to data with error bars

后端 未结 3 2324
旧时难觅i
旧时难觅i 2021-02-20 16:17

I am currently using numpy.polyfit(x,y,deg) to fit a polynomial to experimental data. I would however like to fit a polynomial that uses weighting based on the errors of the poi

相关标签:
3条回答
  • 2021-02-20 16:43

    Take a look at http://scipy-cookbook.readthedocs.io/items/FittingData.html in particular the section 'Fitting a power-law to data with errors'. It shows how to use scipy.optimize.leastsq with a function that includes error weighting.

    0 讨论(0)
  • 2021-02-20 16:43

    For weighted polynomial fitting you can use:

    numpy.polynomial.polynomial.polyfit(x, y, deg, rcond=None, full=False, w=weights)
    

    see http://docs.scipy.org/doc/numpy/reference/generated/numpy.polynomial.polynomial.polyfit.html

    Important to note that in this function the weights should not be supplied as 1/variance (which is the usual form in many weighted applications), but as 1/sigma

    Although curve_fit and leastsq are much more general and powerful optimization tools than polyfit (in that they can fit just any function), polyfit has the advantage that it yields an (exact) analytical solution and is therefore probably much faster than iterative approximation methods like curve_fit and leastsq - especially in the case of fitting polynomials to multiple sets of y-data (obtained at the same x-vector)

    Update: As of numpy version 1.7, numpy.polyfit also takes weights as an input (which ideally should be supplied as 1/sigma, not 1/variance)

    0 讨论(0)
  • 2021-02-20 16:43

    Here is how I did it, with lots of comments!

    Note: I did it with qth and nth order polynomial fits.

    from numpy import *
    import pylab
    
    # get data
    fn = 'cooltemp.dat'
    x, y, xerr, yerr = loadtxt(fn,unpack=True, usecols=[0,1,2,3])
    
    # create nth degree polynomial fit
    n = 1
    zn = polyfit(x,y,n) 
    pn = poly1d(zn) # construct polynomial 
    
    # create qth degree polynomial fit
    q = 5
    zq = polyfit(x,y,q) 
    pq = poly1d(zq)
    
    # plot data and fit
    xx = linspace(0, max(x), 500)
    pylab.plot(xx, pn(xx),'-g', xx, pq(xx),'-b')
    pylab.errorbar(x, y, xerr, yerr, fmt='r.')
    
    # customise graph
    pylab.legend(['degree '+str(n),'degree '+str(q),'data'])
    pylab.axis([0,max(x),0,max(y)])
    pylab.xlabel('x label (unit)')
    pylab.ylabel('y label (unit)')
    
    pylab.show()
    
    0 讨论(0)
提交回复
热议问题