Python实现最小二乘法拟合直线（求斜率截距）

利用最小二乘法拟合直线，实现了对一系列点拟合出其最接近的直线，并给出公式，包括斜率和截距。并且绘制出最终拟合线。

完整代码如下：

# 核心代码，求斜率w,截距b
def fit(data_x, data_y):
    m = len(data_y)
    x_bar = np.mean(data_x)
    sum_yx = 0
    sum_x2 = 0
    sum_delta = 0
    for i in range(m):
        x = data_x[i]
        y = data_y[i]
        sum_yx += y * (x - x_bar)
        sum_x2 += x ** 2
    # 根据公式计算w
    w = sum_yx / (sum_x2 - m * (x_bar ** 2))

    for i in range(m):
        x = data_x[i]
        y = data_y[i]
        sum_delta += (y - w * x)
    b = sum_delta / m
    return w, b
# 模拟数据
x = np.arange(1, 17, 1)
y = np.array([4.00, 6.40, 8.00, 8.80, 9.22, 9.50, 9.70, 10.86, 10.00, 10.20, 10.32, 10.42, 10.50, 11.55, 12.58, 13.60])
# 计算并绘制
w, b = fit(x, y)
pred_y = w * x + b
plt.scatter(x, y)
plt.plot(x, pred_y, c='r', label='line')
plt.show()

计算后绘制的图形如下图所示。 蓝色为原始点数据，红色为拟合线。

原文链接：
https://www.cnblogs.com/arli/p/11428205.html