scubamut
10/12/2014 - 11:15 AM

Fit line to points

Fit line to points

x = np.array([0.0, 1.0, 2.0, 3.0,  4.0,  5.0])
y = np.array([0.0, 0.8, 0.9, 0.1, -0.8, -1.0])
z = np.polyfit(x, y, 1)
yy = [z[0] * x[i] + z[1] for i in range(len(x))]
plt.plot(x,y,'bo',x,yy)