python - 从散点图拟合数据

Question

我有一个包含两列的数据框，我将其散点图绘制成如下图所示:



我想知道是否有办法找到最适合它的分布曲线，因为我发现教程只关注一个变量的分布(例如 this case 。我正在寻找这样的东西:



有没有人对此案例有任何说明或示例代码？

Answer 1

您可以尝试使用 numpy.polyfit 拟合不同次数的多项式.它以 x、y 和拟合多项式的次数作为输入。

您可以编写一个循环，从 1 迭代到 5 度。使用函数返回的系数绘制 f(x)。

对于 d 度数:

使用 np.polyfit(x, y, d) 拟合

获取系数并可选择绘制 f(x) 的度数 d

然后求平方和 (yi - f(xi))^2

请注意，平方和只是误差的一个指示——通常它会随着度数的增加而下降，但绘图会告诉你是否过度拟合数据。

这只是解决问题的方法之一。