[lmfit] 24. 減衰高調波発振器モデルによるカーブフィッティング

はじめに

lmfitは非線形最小二乗法を用いてカーブフィットするためのライブラリであり、Scipy.optimize.curve_fitの拡張版に位置する。ここでは、データを減衰発振器モデルによりカーブフィッティングする方法について説明する。

\[ f(x; A, \mu, \sigma) = \frac{A}{\sqrt{ [1 – (x/\mu)^2]^2 + (2\sigma x/\mu)^2}}\]

上記の関数でデータを作成する。作成したデータは下のようになる。

lmfit.models の DampedOscillatorModelをモデル関数として用いる。

model.guess(y, x=x)により、初期値を推定する。

out = damp_mod.fit(y, pars, x=x)により、カーブフィッティングを実行する。

print(result.fit_report())により、フィッティングの結果を見ることができる。

result.plot_fit()によりデータとフィッティングカーブが表示される。

result.plot()とすることで残差とともにフィッティング結果が表示される。