python - 如何在 Python 中测量最佳拟合线的质量？

Question

我有一些数据点要绘制，并想在图表中添加一条最佳拟合线，然后输出相关指标以指示最佳拟合线的质量。

我可以绘制数据，并且polyfit是我用来添加最佳拟合线的函数。但是，我只是想知道如何获得指示最佳拟合线质量的指标？

我没有看到polyfit返回任何指标（即最小平方误差值）。

数据：

0,1717
1,1761
2,1961
3,1711
4,1285
5,976
6,721
7,428
8,313
9,297
10,375
11,521
12,678
13,752
14,728
15,758
16,741
17,812
18,845
19,863
20,933
21,1169
22,1523
23,1779

Answer 1

均方误差是一种衡量标准。

如果您的行和数据上有每个点的 numpy 数组：

numpy.mean((data - line_vals) ** 2)

编辑：要获得 line_vals，如果您有一个方程 y=mx+b 表示一行：

line_vals = b + m * numpy.linspace(0, 23, 24)