python - python中散点图的最佳拟合直线的代码

Question

下面是我在文本文件中散点绘制数据的代码。我打开的文件包含两列。左列是 x 坐标，右列是 y 坐标。该代码创建了 x 与 y 的散点图。我需要一个代码来绘制一条最适合散点图中数据的线，并且没有一个内置的 pylab 函数对我有用。

from matplotlib import *
from pylab import *

with open('file.txt') as f:
   data = [line.split() for line in f.readlines()]
   out = [(float(x), float(y)) for x, y in data]
for i in out:
   scatter(i[0],i[1])
   xlabel('X')
   ylabel('Y')
   title('My Title')
show()

Answer 1

this excellent answer 的单行版本绘制最佳拟合线是:

plt.plot(np.unique(x), np.poly1d(np.polyfit(x, y, 1))(np.unique(x)))

使用 np.unique(x) 而不是 x 处理 x 未排序或具有重复值的情况。