python - 在 Python/Seaborn 的图例中显示置信区间

Question

我在 Python 上使用 seaborn 和 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')

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

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

Answer 1

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

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

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()