在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')

以下是我目前得到的输出结果：

请问有人知道我应该如何做吗？提前感谢您的帮助。

- Alex Ash

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

- gmds