如何在Python中绘制最佳拟合线

Question

我真的很困惑：I ....我有大量的数据，我正在尝试以最佳拟合线对其进行绘制。 我尝试了两种不同的方法：

pl.plot(med[::skip],var[::skip],'k.')
p, q = np.polyfit(var[::skip],med[::skip], 1)
pl.plot(med,p*med+q,'-')

和

pl.plot(med[::skip],var[::skip],'k.')
p = np.polyfit(var[::skip],med[::skip], 1)
fit = np.polyval(p, var[::skip])
pl.plot(var[::skip],fit)

但它们都给我疯狂的东西：

我究竟做错了什么？

Answer 1

numpy.polyfit()将x然后y作为其参数，因此您需要在调用它时交换var和med 。

请注意，因为您具有对数-对数图，所以这不会给您一条直线。 相反，您应该适合两个变量的对数：

pl.plot(med[::skip],var[::skip],'k.')
p, q = np.polyfit(np.log10(med[::skip]),np.log10(var[::skip]), 1)
pl.plot(med[::skip],10**(p*np.log10(med[::skip])+q),'-')