如何将趋势线添加到使用matplotlib.scatter绘制的点图?
答案 0 :(得分:50)
解释here
在numpy的帮助下,可以计算出一个线性拟合。
# 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])