最小二乘法相關(guān)知識，我把他分解成了四篇文章，第一篇是推導(dǎo)過 程 ，第二篇是 系數(shù)求解 ，第三篇是C語言實現(xiàn)二乘法（二次曲線）。有興趣可以看文章底部，往期推薦。

這篇文章是第四篇，主要是用Python驗證C語言實現(xiàn)二乘法的正確性，因為網(wǎng)絡(luò)上的文章錯綜復(fù)雜，復(fù)制粘貼有風險，需要保證功能的正確性。

相關(guān)代碼如下：

复制# -*- coding: utf-8 -*-
import numpy as np
#import random
import matplotlib.pyplot as plt
from scipy.optimize import curve_fit

#自定義函數(shù)
def func(x, a, b, c):
    return a + b*x + c*pow(x, 2)

#定義x、y散點坐標
x = np.array([0,2,4,6,8,10])
y = np.array([0,9,24,42,76,118])
#非線性最小二乘法擬合
popt, pcov = curve_fit(func, x, y)
#獲取popt里面是擬合系數(shù)
a = popt[0] 
b = popt[1]
c = popt[2]
yvals = func(x, a, b, c) #擬合y值
print('a:', a)
print('b:', b)
print('c:', c)
#繪圖
plot1 = plt.plot(x, y, 's') #square
plot2 = plt.plot(x, yvals, 'r')
plt.xlabel('x')
plt.ylabel('y')
plt.show()

上述代碼運行結(jié)果如下圖：

其中，藍色的矩形點是已知的坐標點，而紅色的則是對已知坐標點擬合而成二次曲線。