python - 使用对数对数图进行多项式拟合

Question

我有一个简单的问题，需要在对数刻度上拟合一条直线。我的代码是，

data=loadtxt(filename)
xdata=data[:,0]
ydata=data[:,1]
polycoeffs = scipy.polyfit(xdata, ydata, 1)
yfit = scipy.polyval(polycoeffs, xdata)
pylab.plot(xdata, ydata, 'k.')
pylab.plot(xdata, yfit, 'r-')

现在我需要在对数刻度上绘制拟合线，所以我只需更改 x 和 y 轴，

ax.set_yscale('log')
ax.set_xscale('log')

那么它就没有绘制正确的拟合线。那么如何更改拟合函数(以对数刻度)，以便它可以在对数刻度上绘制拟合线？

Answer 1

编辑:

from scipy import polyfit
data = loadtxt("data.txt")
xdata,ydata = data[:,0],data[:,1]
xdata,ydata = zip(*sorted(zip(xdata,ydata))) # sorts the two lists after the xdata    

xd,yd = log10(xdata),log10(ydata)
polycoef = polyfit(xd, yd, 1)
yfit = 10**( polycoef[0]*xd+polycoef[1] )

plt.subplot(211)
plt.plot(xdata,ydata,'.k',xdata,yfit,'-r')
plt.subplot(212)
plt.loglog(xdata,ydata,'.k',xdata,yfit,'-r')
plt.show()