在python中使用lmfit模块，如何调用模型中的参数？

Question

import numpy as np

import matplotlib.pyplot as plt

from lmfit import minimize, Parameters, Parameter, report_fit


# create data to be fitted


x = np.linspace(0, 15, 301)

data = (5. * np.sin(2 * x - 0.1) * np.exp(-x*x*0.025) +

np.random.normal(size=len(x), scale=0.2) )

# define objective function: returns the array to be minimized


def fcn2min(params, x, data):

    """ model decaying sine wave, subtract data"""
    amp = params['amp'].value
    shift = params['shift'].value
    omega = params['omega'].value
    decay = params['decay'].value
    model = amp * np.sin(x * omega + shift) * np.exp(-x*x*decay)


    return model - data


# create a set of Parameters

params = Parameters()

params.add('amp',   value= 10,  min=0)

params.add('decay', value= 0.1)

params.add('shift', value= 0.0, min=-np.pi/2., max=np.pi/2)

params.add('omega', value= 5.0)


# do fit, here with leastsq model

result = minimize(fcn2min, params, args=(x, data))


# calculate final result

final = data + result.residual


# try to plot results
plt.plot(x,data,'k+')

plt.plot(x,final,'r')

plt.show()

在这段代码中，我想在python中调用'amp'，'shift'等参数。 打印（amp）..各种东西 拟合后如何在python中调用这些参数？ 当我使用 print(amp) 时，会显示错误消息；名称“amp”未定义。如何使用打印功能打印这些拟合参数？ （等打印（放大器））

Answer 1

您可能试图在函数之外打印该数据。 amp、shift、omega 和 decay 变量位于 fc2min 的本地范围内，因此只能在函数内部访问。你的数据分析技能似乎远远超过你的 Python 知识，所以我在这段代码中添加了一些有用的提示：

import numpy as np
import matplotlib.pyplot as plt
from lmfit import minimize, Parameters, Parameter, report_fit

# create data to be fitted
x = np.linspace(0, 15, 301)
data = (5. * np.sin(2 * x - 0.1) * np.exp(-x*x*0.025) +
np.random.normal(size=len(x), scale=0.2) )

# define objective function: returns the array to be minimized
def fcn2min(params, x, data):

    """ model decaying sine wave, subtract data"""
    amp = params['amp'].value
    shift = params['shift'].value
    omega = params['omega'].value
    decay = params['decay'].value
    model = amp * np.sin(x * omega + shift) * np.exp(-x*x*decay)

    # tell Python we're modifying the model_data list
    # that was declared outside of this function
    global model_data

    # store the model data produced by this function call
    # add any data you want to display later to this "dictionary"
    model_data += [{
        "amp": amp,
        "shift": shift,
        "omega": omega,
        "decay": decay
    }]

    return model - data


# create a set of Parameters
params = Parameters()
params.add('amp',   value= 10,  min=0)
params.add('decay', value= 0.1)
params.add('shift', value= 0.0, min=-np.pi/2., max=np.pi/2)
params.add('omega', value= 5.0)

# declare an empty list to hold the model data
model_data = []

# do fit, here with leastsq model
result = minimize(fcn2min, params, args=(x, data))

# print each item in the model data list
for datum in model_data:
    for key in datum:

        #the 5 in %.5f controls the precision of the floating point value
        print("%s: %.5f  ") % (key, datum[key]),
    print


# calculate final result
final = data + result.residual

# try to plot results
plt.plot(x,data,'k+')
plt.plot(x,final,'r')
plt.show()