[Numpy-discussion] Line of best fit!

Scott Sinclair scott.sinclair.za at gmail.com
Tue Dec 9 00:13:26 EST 2008


> 2008/12/9 Angus McMorland <amcmorl at gmail.com>:
> Hi James,
>
> 2008/12/8 James <james at fmnmedia.co.uk>:
>>
>> I have a very simple plot, and the lines join point to point, however i
>> would like to add a line of best fit now onto the chart, i am really new
>> to python etc, and didnt really understand those links!
>>
>> Can anyone help me :)
>
> It sounds like the second link, about linear regression, is a good
> place to start, and I've made a very simple example based on that:
>
> -----------------------------------------------
> import numpy as np
> import matplotlib.pyplot as plt
>
> x = np.linspace(0, 10, 11) #1
> data_y = np.random.normal(size=x.shape, loc=x, scale=2.5) #2
> plt.plot(x, data_y, 'bo') #3
>
> coefs = np.lib.polyfit(x, data_y, 1) #4
> fit_y = np.lib.polyval(coefs, x) #5
> plt.plot(x, fit_y, 'b--') #6
> ------------------------------------------------

James, you'll want to add an extra line to the above code snippet so
that Matplotlib displays the plot:

plt.show()

Cheers,
Scott



More information about the NumPy-Discussion mailing list