Jak dodać linię trendu w wykresach punktowych python matplotlib (scatter)?

Question

Jak mogę dodać linię trendu do wykresu punktowego narysowanego za pomocą matplotlib.scatter?

Answer 1

jak wyjaśniono here

Z pomocą numpy można obliczyć na przykład liniowe dopasowanie.

# plot the data itself 
pylab.plot(x,y,'o') 

# calc the trendline 
z = numpy.polyfit(x, y, 1) 
p = numpy.poly1d(z) 
pylab.plot(x,p(x),"r--") 
# the line equation: 
print "y=%.6fx+(%.6f)"%(z[0],z[1])