gpt4 book ai didi

python - 曲线拟合数据问题

转载 作者:太空宇宙 更新时间:2023-11-04 03:23:01 24 4
gpt4 key购买 nike

import numpy as np
import matplotlib.pyplot as plt
from scipy.optimize import curve_fit

xdata = np.array([177,180,183,187,189,190,196,197,201,202,203,204,206,218,225,231,234,
252,262,266,267,268,277,286,303])

ydata = np.array([0.81,0.74,0.78,0.75,0.77,0.81,0.73,0.76,0.71,0.74,0.81,0.71,0.74,0.71,
0.72,0.69,0.75,0.59,0.61,0.63,0.64,0.63,0.35,0.27,0.26])



def f(x, n1, n2, n3, n4, n5):
if (n1 > 0.2 and n1 < 0.8 and
n2 > -0.3 and n2 < 0):
return n1 + (n2 * x + n3) * 1./ (1 + np.exp(n4 * (n5 - x)))
return 1e38

coeffs, pcov = curve_fit(f, xdata, ydata, p0 = (0.29, -0.005, 1.0766, -0.36397, 104))

ploty = f(xdata, coeffs[0], coeffs[1], coeffs[2], coeffs[3], coeffs[4])
for i in range(1, len(coeffs) + 1):
print ('n%s = %s' % (i, coeffs[i - 1]))

无法正常工作并出现此警告:

OptimizeWarning: Covariance of the parameters could not be estimated
category=OptimizeWarning)

但对

正常工作
xdata = np.array([73.0, 80.0, 88.0, 93.8, 96.3, 98.5, 100.0, 101.0, 102.3,  104.0, 105.3,
107.0, 109.5, 111.5, 114.0, 117.0, 119.5, 121.0, 124.0])
ydata = np.array([0.725, 0.7, 0.66, 0.63, 0.615, 0.61, 0.59, 0.56, 0.53, 0.49, 0.45,
0.41, 0.35, 0.32, 0.3, 0.29, 0.29, 0.29, 0.29])

Lmfit 也不起作用。

最佳答案

由于您还使用了 lmfit 作为标签,这里是一个使用 lmfit 的解决方案.您获得的参数值如下所示:

n1:   0.26564921 +/- 0.024765 (9.32%) (init= 0.2)
n2: -0.00195398 +/- 0.000311 (15.93%) (init=-0.005)
n3: 0.87261892 +/- 0.068601 (7.86%) (init= 1.0766)
n4: -1.43507072 +/- 1.223086 (85.23%) (init=-0.36379)
n5: 277.684530 +/- 3.768676 (1.36%) (init= 274)

导致以下输出: enter image description here

如您所见,拟合很好地再现了数据,并且参数在要求的范围内;您的函数中不需要 if 语句。

这是重现情节的完整代码,并附有一些额外的注释:

from lmfit import minimize, Parameters, Parameter, report_fit
import numpy as np

xdata = np.array([177.,180.,183.,187.,189.,190.,196.,197.,201.,202.,203.,204.,206.,218.,225.,231.,234.,
252.,262.,266.,267.,268.,277.,286.,303.])

ydata = np.array([0.81,0.74,0.78,0.75,0.77,0.81,0.73,0.76,0.71,0.74,0.81,0.71,0.74,0.71,
0.72,0.69,0.75,0.59,0.61,0.63,0.64,0.63,0.35,0.27,0.26])

def fit_fc(params, x, data):
n1 = params['n1'].value
n2 = params['n2'].value
n3 = params['n3'].value
n4 = params['n4'].value
n5 = params['n5'].value

model = n1 + (n2 * x + n3) * 1./ (1. + np.exp(n4 * (n5 - x)))

return model - data #that's what you want to minimize

# create a set of Parameters
# 'value' is the initial condition
# 'min' and 'max' define your boundaries
params = Parameters()
params.add('n1', value= 0.2, min=0.2, max=0.8)
params.add('n2', value= -0.005, min=-0.3, max=10**(-10))
params.add('n3', value= 1.0766, min=-1000., max=1000.)
params.add('n4', value= -0.36379, min=-1000., max=1000.)
params.add('n5', value= 274.0, min=0., max=1000.)

# do fit, here with leastsq model
result = minimize(fit_fc, params, args=(xdata, ydata))

# write error report
report_fit(params)

xplot = np.linspace(min(xdata), max(xdata), 1000)
yplot = result.values['n1'] + (result.values['n2'] * xplot + result.values['n3']) * \
1./ (1. + np.exp(result.values['n4'] * (result.values['n5'] - xplot)))
#plot results
try:
import pylab
pylab.plot(xdata, ydata, 'k+')
pylab.plot(xplot, yplot, 'r')
pylab.show()
except:
pass

编辑:

事实证明,这只适用于 0.8.3 版本。如果你使用 0.9.x 版本,你需要相应地调整你的代码;检查here从 0.8.3 到 0.9.x 进行了哪些更改。

关于python - 曲线拟合数据问题,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/34084635/

24 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com