使用lmfit拟合两个变量

Question

我如何在python中使用lmfit将这个函数与两个输入变量拟合？函数f（x）= a *（x-b）** 2，a和b是变量，它们可以是随机数。

Answer 1

您可以尝试这个吗？

import matplotlib.pyplot as plt
import numpy as np
from lmfit.models import ExpressionModel

# synthetic data
x = np.linspace(-10, 10, 100)
a,b = 0.9,1.9
y = a*(x-b)**2

# fit y using lmfit; with "guesses": a=1, b=2
gmod = ExpressionModel("a*(x-b)**2")
result = gmod.fit(y, x=x, a=1, b=2)

print(result.fit_report())

plt.plot(x, y, 'bo')
plt.plot(x, result.init_fit, 'k--', label='initial fit')
plt.plot(x, result.best_fit, 'r-', label='best fit')
plt.legend(loc='best')
plt.show()

输出：