在loglog规模上使用matplotlib拟合曲线

Question

我在python中使用loglog函数绘制简单的2D图,如下所示:

plt.loglog(x,y,label='X vs Y');

X和Y都是n大小浮动数的列表.

我想在同一个图表上插入一条线.我试过numpy.polyfit,但我无处可去.

如果您的图表已经是loglog规模,那么如何使用polyfit来匹配一条线？

Answer 1

Numpy并不关心matplotlib图的轴是什么.

我认为你认为log(y)是一些多项式函数log(x),你想找到那个多项式？如果是这种情况,那么运行numpy.polyfit数据集的对数:

import numpy as np
logx = np.log(x)
logy = np.log(y)
coeffs = np.polyfit(logx,logy,deg=3)
poly = np.poly1d(coeffs)

poly现在是一个log(x)返回的多项式log(y).为了适应预测y值,您可以定义一个只对多项式求幂的函数:

yfit = lambda x: np.exp(poly(np.log(x)))

您现在可以在matplotlib loglog图上绘制拟合线:

plt.loglog(x,yfit(x))

并像这样展示它

plt.show()