[lmfit] 17. フィッティングして得られたパラメータをDataFrameに変換する

はじめに

lmfitは非線形最小二乗法を用いてカーブフィットするためのライブラリであり、Scipy.optimize.curve_fitの拡張版に位置する。ここでは、lmfitによるカーブフィッティングで得られたパラメータをpandasのDataFrameに変換する方法について説明する。

yデータはrng = default_rng()とし、rng.normal()でガウス分布に従うデータを10000こ作成し、ヒストグラムにすることで準備した。xとyの関係を図で示すと以下のようになる。

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

model.guess(y, x=x)により、上図のデータをsin関数モデルで近似するためのフィッティングパラメータについて、初期値を推定する。パラメータ(params)は以下のようになる。

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

result.plot_fit()によりデータとフィッティングカーブが表示される。
しかし、これだとただの折れ線グラフに見えるので、フィッティングパラメータを得て、細かいステップでガウス分布曲線を描写する。

フィッティングで得られたパラメータはresult.paramsで見ることができる。

これをvaluesdictによって、順序つき辞書に変換する。

辞書をDataFrameにするには、pd.DataFrame.from_dict()を用いる。

以下のようなDataFrameが得られる。