利用Python進行多項式擬合

多項式擬合的簡單代碼:

import matplotlib.pyplot as plt
import numpy as np

x=[1,2,3,4,5,6,7,8]
y=[1,4,9,13,30,25,49,70]

a=np.polyfit(x,y,2)   #用2次多項式擬合x,y數組
b=np.poly1d(a)        #擬合完成后生成多項式對象
c=b(x)                #獲取x在多項式處的值

plt.scatter(x,y,marker='o',label='original datas')                  #對原始數據做散點圖
plt.plot(x,c,ls='--',c='red',label='fitting with second polynomial')#對擬合之后的數據作圖
plt.legend()                                                        #給圖加上圖例
plt.show()

結果: