seabornのlmplot(多項式回帰)について

Question

sns.lmplotのorderで多項式回帰を可視化したときの式を得る方法はないのでしょうか。
調べ方が悪いのか全然見つかりません。
もし、方法があるのなら教えて欲しいです。

Answer 1

ソースコード seaborn/regression.py を見ると、numpy.polyfit で多項式回帰しているので、sns.lmplot() の結果の係数を取り出すことはできないですが、numpy.polyfit を直接呼び出せば、係数を取得できます。

import numpy as np
import pandas as pd
import seaborn as sns

x = np.linspace(-3, 3, 100)
df = pd.DataFrame({"x": x, "y": x ** 2 + np.random.randn(len(x))})

tips = sns.load_dataset("tips")

g = sns.lmplot(x="x", y="y", data=df, order=2)

coef = np.polyfit(df["x"], df["y"], 2)
print(coef)

a1, a2, a3 = coef
# order=2 の場合、これが質問で求めている式
f = lambda x: a1 * x ** 2 + a2 * x + a3

# seaborn の結果と一致することを確認
g.ax.plot(x, f(x), "r")