python - 方程式と R2 テキストを使用した Seaborn のインプロット

Question

通常のデータ分析作業では、seaborn パッケージが利用可能になったので、100% python を使用するように切り替えました。この素晴らしいパッケージに感謝します。ただし、lmplot() 関数を使用すると、polyfit 方程式や R2 値を表示する機能が欠けています。それを追加する簡単な方法を知っている人はいますか？

Answer 1

複数の回帰適合がある場合 (つまり、または変数lmplotを使用する場合) に対応する値が定義されていないため、 で自動的に実行することはできません。huerowcol

しかし、これは同様のjointplot機能の一部です。デフォルトでは、相関係数と p 値が表示されます。

import seaborn as sns
import numpy as np

x, y = np.random.randn(2, 40)
sns.jointplot(x, y, kind="reg")

ただし、任意の関数を渡すことができます。R^2 が必要な場合は、次のようにします。

from scipy import stats
def r2(x, y):
    return stats.pearsonr(x, y)[0] ** 2
sns.jointplot(x, y, kind="reg", stat_func=r2)