誰能幫我解決 curve.fit 的擬合問題。我想將我的數據擬合到二階方程。但我得到了像線性方程一樣的結果。這是我的代碼：import numpy as npimport matplotlib.pyplot as pltfrom scipy.optimize import curve_fitdef func(x, a, b, c): f = a*np.power(x, 2) + b*x + c return fxdata_prime=[3.0328562996216282, 3.101784841139168, 3.1707134502066894, 3.2396419917242292, 3.308570533241769, 3.3774990747593088, 3.3774990747593088, 3.4337789932367149, 3.4900589392912855, 3.5463388577686916, 3.6026187762460977, 3.6588987223006684]ydata_prime=[6.344300000000002, 6.723900000000002, 7.080399999999999, 7.399800000000001, 7.649099999999999, 7.753100000000002, 7.753100000000002, 7.658600000000002, 7.442100000000002, 7.180100000000001, 6.902700000000001, 6.6211]plt.plot(xdata_prime, ydata_prime, 'b-', label='data')popt, pcov = curve_fit(func, xdata_prime, ydata_prime)poptplt.plot(xdata_prime, func(xdata_prime, *popt), 'r-',label='fit')plt.xlabel('x')plt.ylabel('y')plt.legend()plt.show()
1 回答
青春有我
TA貢獻1784條經驗 獲得超8個贊
您的數組需要是 numpy 數組，因為您的函數正在執行向量化操作（即 a*np.power(x, 2)）。所以這樣你的代碼就可以工作了：
import numpy as np
import matplotlib.pyplot as plt
from scipy.optimize import curve_fit
def func(x, a, b, c):
f = a*np.power(x, 2) + b*x + c
return f
xdata_prime=np.array([3.0328562996216282, 3.101784841139168, 3.1707134502066894, 3.2396419917242292, 3.308570533241769, 3.3774990747593088, 3.3774990747593088, 3.4337789932367149, 3.4900589392912855, 3.5463388577686916, 3.6026187762460977, 3.6588987223006684])
ydata_prime=np.array([6.344300000000002, 6.723900000000002, 7.080399999999999, 7.399800000000001, 7.649099999999999, 7.753100000000002, 7.753100000000002, 7.658600000000002, 7.442100000000002, 7.180100000000001, 6.902700000000001, 6.6211])
plt.plot(xdata_prime, ydata_prime, 'b-', label='data')
popt, pcov = curve_fit(func, xdata_prime, ydata_prime)
plt.plot(xdata_prime, func(xdata_prime, *popt), 'r-',label='fit')
plt.xlabel('x')
plt.ylabel('y')
plt.legend()
plt.show()
添加回答
舉報
0/150
提交
取消