1. 程式人生 > >python numpy 多項式擬合

python numpy 多項式擬合

#ecoding=utf-8
import matplotlib.pyplot as plt
#生成離散x
x = np.arange(-10 , 10 , 2)
#生成離散y
y = -2*x**2 - 5*x + 3
#二項式擬合
pfit = np.polyfit(x , y , 2)
y_fun = np.poly1d(pfit)

plt.plot(x, y,'*', color='r' )
plt.plot(x , y_fun(x) , color='g')

plt.show()
pfit = np.polyfit(x , y , 2)   #根據已有點訓練得到多項式引數 , 三個引數分別為x , y ,多項式最高項次數
y_fun = np.poly1d(pfit)        #得到多項式的方程物件
y = y_fun(x)                   #使用方程