Подгонка двух переменных с использованием lmfit

Question

Как мне связать эту функцию с двумя входными переменными, используя lmfit в python? function 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()

Вывод: