组通过散点图超级并处回归线和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

添加到情节