組通過散點圖超級並處迴歸線和RSQ

Question

我有一個數據幀，我想繪製每個宇宙的次要情節在我的數據

我這樣做產生的散點圖以下罰款。輸出是每個宇宙的散點圖，其中csk在y軸上，msk在x軸上。我正試圖在每個子圖上疊加回歸直線，並在每個子圖上添加Rsq。有沒有辦法做到這一點？被seaborn更好地使用（「regplot」）

plt.suptitle(mytitle, fontsize = 20) 
for (universe, group), ax in zip(temp1.groupby([temp1.universe], sort = False), axes.flatten()): 
    group.plot(x='msk', y='csk', kind='scatter', ax=ax, title=universe,s=2) 
    ax.set_xlim(0,1) 
    ax.set_ylim(0,1) 
    ax.set_xlabel("msk", fontsize =10) 
    ax.set_ylabel("csk", fontsize = 10) 
    ax.set_title(universe, fontsize = 12) 

Answer 1

擬合迴歸線可以用scipy.optimize.curve_fit或scipy.stats.linregress很簡單地完成。

使用這些函數的輸出，您可以很容易地用matplotlib的ax.plot函數繪製迴歸圖。

計算R^2從您的數據和擬合（查找公式，如果你不知道它，它很簡單），然後將其與ax.annotate或ax.text

添加到情節