1. 程式人生 > >python中matplotlib實現最小二乘法擬合的過程詳解

python中matplotlib實現最小二乘法擬合的過程詳解

ast array plt atp ons 正則 key code 擬合

這篇文章主要給大家介紹了關於python中matplotlib實現最小二乘法擬合的相關資料,文中通過示例代碼詳細介紹了關於最小二乘法擬合直線和最小二乘法擬合曲線的實現過程,需要的朋友可以參考借鑒,下面來一起看看吧。

前言

最小二乘法Least Square Method,做為分類回歸算法的基礎,有著悠久的歷史(由馬裏·勒讓德於1806年提出)。它通過最小化誤差的平方和尋找數據的最佳函數匹配。利用最小二乘法可以簡便地求得未知的數據,並使得這些求得的數據與實際數據之間誤差的平方和為最小。最小二乘法還可用於曲線擬合。其他一些優化問題也可通過最小化能量或最大化熵用最小二乘法來表達。

下面這篇文章主要跟大家介紹了關於python中matplotlib實現最小二乘法擬合的相關內容,下面話不多說,來一起看看詳細的介紹:

一、最小二乘法擬合直線

生成樣本點

首先,我們在直線 y = 3 + 5x 附近生成服從正態分布的隨機點,作為擬合直線的樣本點。

?
1 2 3 4 5 6 7 8 9 10 import numpy as np import matplotlib.pyplot as plt # 在直線 y = 3 + 5x 附近生成隨機點 X = np.arange(0, 5, 0.1) Z = [3 + 5 * x for x in X] Y = [np.random.normal(z, 0.5) for z in Z] plt.plot(X, Y, ‘ro‘)
plt.show()

樣本點如圖所示:

技術分享圖片

擬合直線

設 y = a0 + a1*x,我們利用最小二乘法的正則方程組來求解未知系數 a0 與 a1。

技術分享圖片

numpy 的 linalg 模塊中有一個 solve 函數,它可以根據方程組的系數矩陣和方程右端構成的向量來求解未知量。

?
1 2 3 4 5 6 7 8 9 10 11 12 13 def linear_regression(x, y): N = len(x) sumx = sum(x) sumy = sum(y) sumx2 = sum(x**2) sumxy = sum(x*y)
A = np.mat([[N, sumx], [sumx, sumx2]]) b = np.array([sumy, sumxy]) return np.linalg.solve(A, b) a0, a1 = linear_regression(X, Y)

繪制直線

此時,我們已經得到了擬合後的直線方程系數 a0 和 a1。接下來,我們繪制出這條直線,並與樣本點做對比。

?
1 2 3 4 5 6 7 # 生成擬合直線的繪制點 _X = [0, 5] _Y = [a0 + a1 * x for x in _X] plt.plot(X, Y, ‘ro‘, _X, _Y, ‘b‘, linewidth=2) plt.title("y = {} + {}x".format(a0, a1)) plt.show()

擬合效果如下:

技術分享圖片

二、最小二乘法擬合曲線

生成樣本點

與生成直線樣本點相同,我們在曲線 y = 2 + 3x + 4x^2 附近生成服從正態分布的隨機點,作為擬合曲線的樣本點。

?
1 2 3 4 5 6 7 8 9 10 import numpy as np import matplotlib.pyplot as plt # y = 2 + 3x + 4x^2 X = np.arange(0, 5, 0.1) Z = [2 + 3 * x + 4 * x ** 2 for x in X] Y = np.array([np.random.normal(z,3) for z in Z]) plt.plot(X, Y, ‘ro‘) plt.show()

樣本點如圖所示:

技術分享圖片

擬合曲線

設該曲線的方程為 y = a0 + a1*x + a2*x^2,同樣,我們通過正則方程組來求解未知量 a0、a1 和 a2。

技術分享圖片

?
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 # 生成系數矩陣A def gen_coefficient_matrix(X, Y): N = len(X) m = 3 A = [] # 計算每一個方程的系數 for i in range(m): a = [] # 計算當前方程中的每一個系數 for j in range(m): a.append(sum(X ** (i+j))) A.append(a) return A # 計算方程組的右端向量b def gen_right_vector(X, Y): N = len(X) m = 3 b = [] for i in range(m): b.append(sum(X**i * Y)) return b A = gen_coefficient_matrix(X, Y) b = gen_right_vector(X, Y) a0, a1, a2 = np.linalg.solve(A, b)

繪制曲線

我們根據求得的曲線方程,繪制出曲線的圖像。

?
1 2 3 4 5 6 7 # 生成擬合曲線的繪制點 _X = np.arange(0, 5, 0.1) _Y = np.array([a0 + a1*x + a2*x**2 for x in _X]) plt.plot(X, Y, ‘ro‘, _X, _Y, ‘b‘, linewidth=2) plt.title("y = {} + {}x + {}$x^2$ ".format(a0, a1, a2)) plt.show()

擬合效果如下:

技術分享圖片

總結

以上就是這篇文章的全部內容了,希望本文的內容對大家的學習或者工作能帶來一定的幫助,如果有疑問大家可以留言交流,謝謝大家對腳本之家的支持。

您可能感興趣的文章:

  • Python中實現最小二乘法思路及實現代碼

原文鏈接:http://www.codebelief.com/article/2017/04/matplotlib-demonstrate-least-square-regression-process/

python中matplotlib實現最小二乘法擬合的過程詳解