首页 > 解决方案 > 在 Python / Seaborn 的图例中显示置信区间

问题描述

seaborn我正在使用Python 上的函数生成一些带有线性回归和置信区间的散点图sns.regplot。我可以找到一种在图例中显示回归线的方法,但我还想在图例中添加置信区间(以透明蓝色作为参考颜色)。

这是我拥有的代码和到目前为止我得到的结果。

Tobin_Nationality_Reg = sns.regplot(x="Nationality_Index_Normalized",
                        y="Tobins_Q_2017",
                        data=Scatter_Plot,
                        line_kws={'label':'Regression line'})

plt.xlabel("Nationality Index")
plt.ylabel("Tobin's Q")
plt.legend()`
plt.savefig('Tobin_Nationality_Reg.png')

这是我目前得到的输出:散点图

在此处输入图像描述

有人知道我该怎么做吗?提前致谢。

标签: pythonmatplotlibseaborn

解决方案


我相信没有干净的方法可以做到这一点,因为seaborn不会为fill_between绘制置信区间的调用公开关键字参数。

但是,可以通过直接修改label属性来完成PolyCollection

x, y = np.random.rand(2, 20)

ax = sns.regplot(x, y, line_kws={'label': 'Regression line'})
ax.collections[1].set_label('Confidence interval')
ax.legend()

在此处输入图像描述


推荐阅读