最小二乘法的推導(dǎo)可參考下面這個(gè)博客,推導(dǎo)非常詳細(xì).
https://my.oschina.net/keyven/blog/526010
首先畫一個(gè)加入了噪聲的散點(diǎn)圖,函數(shù)是y=0.2x,加入噪聲后如下圖所示:

代碼如下:

# -*- coding: utf-8 -* import numpy as np import matplotlib.pyplot as plt x = np.linspace(0, 30, num=50) y = 0.2*x+[np.random.random() for _ in range(50)] if __name__ == '__main__':plt.figure(figsize=(10, 5), facecolor='w')plt.plot(x, y, 'ro', lw=2, markersize=6)plt.grid(b=True, ls=':')plt.xlabel(u'X', fontsize=16)plt.ylabel(u'Y', fontsize=16)plt.show()

我將上面這個(gè)代碼保存為paint_scatter_plot.py這個(gè)名字,方便其他地方調(diào)用.
接下來畫出擬合直線:

代碼如下:

from paint_scatter_plot import x,y import numpy as np import matplotlib.pyplot as plt def Least_squares(x,y):x_ = x.mean()y_ = y.mean()m = np.zeros(1)n = np.zeros(1)k = np.zeros(1)p = np.zeros(1)for i in np.arange(50):k = (x[i]-x_)* (y[i]-y_)m += kp = np.square( x[i]-x_ )n = n + pa = m/nb = y_ - a* x_return a,b if __name__ == '__main__':a,b = Least_squares(x,y)print a,by1 = a * x + bplt.figure(figsize=(10, 5), facecolor='w')plt.plot(x, y, 'ro', lw=2, markersize=6)plt.plot(x, y1, 'r-', lw=2, markersize=6)plt.grid(b=True, ls=':')plt.xlabel(u'X', fontsize=16)plt.ylabel(u'Y', fontsize=16)plt.show()

代碼運(yùn)算結(jié)果: a = 0.19784577, b = 0.48503344 ,還是非常接近正確結(jié)果的.

總結(jié)

以上是生活随笔為你收集整理的python最小二乘法拟合直线的全部內(nèi)容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網(wǎng)站內(nèi)容還不錯(cuò)，歡迎將生活随笔推薦給好友。