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)                   #使用方程