您好,登錄后才能下訂單哦!
這篇文章主要介紹python如何對任意數據和曲線進行擬合并求出函數表達式,文中介紹的非常詳細,具有一定的參考價值,感興趣的小伙伴們一定要看完!
具體示例如下。
###擬合年齡
import numpy as np import matplotlib.pyplot as plt #定義x、y散點坐標 x = [10,20,30,40,50,60,70,80] x = np.array(x) print('x is :\n',x) num = [174,236,305,334,349,351,342,323] y = np.array(num) print('y is :\n',y) #用3次多項式擬合 f1 = np.polyfit(x, y, 3) print('f1 is :\n',f1) p1 = np.poly1d(f1) print('p1 is :\n',p1) #也可使用yvals=np.polyval(f1, x) yvals = p1(x) #擬合y值 print('yvals is :\n',yvals) #繪圖 plot1 = plt.plot(x, y, 's',label='original values') plot2 = plt.plot(x, yvals, 'r',label='polyfit values') plt.xlabel('x') plt.ylabel('y') plt.legend(loc=4) #指定legend的位置右下角 plt.title('polyfitting') plt.show()
2 。 第一種方案是給出具體的函數形式(可以是任意的,只要你能寫的出來 下面的func就是),用最小二乘的方式去逼近和擬合,求出函數的各項系數,如下。
##使用curve_fit import numpy as np import matplotlib.pyplot as plt from scipy.optimize import curve_fit #自定義函數 e指數形式 def func(x, a, b,c): return a*np.sqrt(x)*(b*np.square(x)+c) #定義x、y散點坐標 x = [20,30,40,50,60,70] x = np.array(x) num = [453,482,503,508,498,479] y = np.array(num) #非線性最小二乘法擬合 popt, pcov = curve_fit(func, x, y) #獲取popt里面是擬合系數 print(popt) a = popt[0] b = popt[1] c = popt[2] yvals = func(x,a,b,c) #擬合y值 print('popt:', popt) print('系數a:', a) print('系數b:', b) print('系數c:', c) print('系數pcov:', pcov) print('系數yvals:', yvals) #繪圖 plot1 = plt.plot(x, y, 's',label='original values') plot2 = plt.plot(x, yvals, 'r',label='polyfit values') plt.xlabel('x') plt.ylabel('y') plt.legend(loc=4) #指定legend的位置右下角 plt.title('curve_fit') plt.show()
擬合高斯分布的方法。
#encoding=utf-8 import numpy as np import matplotlib.pyplot as plt from scipy.optimize import curve_fit import pandas as pd #自定義函數 e指數形式 def func(x, a,u, sig): return a*(np.exp(-(x - u) ** 2 /(2* sig **2))/(math.sqrt(2*math.pi)*sig))*(431+(4750/x)) #定義x、y散點坐標 x = [40,45,50,55,60,65,70,75,80,85,90,95,100,105,110,115,120,125,130,135] x=np.array(x) # x = np.array(range(20)) print('x is :\n',x) num = [536,529,522,516,511,506,502,498,494,490,487,484,481,478,475,472,470,467,465,463] y = np.array(num) print('y is :\n',y) popt, pcov = curve_fit(func, x, y,p0=[3.1,4.2,3.3]) #獲取popt里面是擬合系數 a = popt[0] u = popt[1] sig = popt[2] yvals = func(x,a,u,sig) #擬合y值 print(u'系數a:', a) print(u'系數u:', u) print(u'系數sig:', sig) #繪圖 plot1 = plt.plot(x, y, 's',label='original values') plot2 = plt.plot(x, yvals, 'r',label='polyfit values') plt.xlabel('x') plt.ylabel('y') plt.legend(loc=4) #指定legend的位置右下角 plt.title('curve_fit') plt.show()
以上是“python如何對任意數據和曲線進行擬合并求出函數表達式”這篇文章的所有內容,感謝各位的閱讀!希望分享的內容對大家有幫助,更多相關知識,歡迎關注億速云行業資訊頻道!
免責聲明:本站發布的內容(圖片、視頻和文字)以原創、轉載和分享為主,文章觀點不代表本網站立場,如果涉及侵權請聯系站長郵箱:is@yisu.com進行舉報,并提供相關證據,一經查實,將立刻刪除涉嫌侵權內容。