Добавить легенду в sns.regplot и sns.lmplot

Question

Я новичок в Саборне и хочу добавить легенду к этому сюжету. Я ничего подобного не нашел, к сожалению.

sns.lmplot(x="x", y="y4", data=data, order=2, scatter_kws={"s":5}, line_kws={"color": "red"}, ci=95)
sns.regplot(x="x", y="y4", data=data, scatter_kws={"s":5}, line_kws={"color": "green"}, )
plt.show();

В качестве легенды я хочу добавить следующее:
синие точки как data,
зеленая линия как y,
красная линия как y^

Как мне решить эту проблему

Answer 1

Я не знаю, как вы создали график рассеяния.Вероятно, используя plt.scatter.Но вот как вы можете изменить легенду для lmplot и regplot как

l = sns.lmplot(x="x", y="y4", data=data, order=2, scatter_kws={"s":5}, line_kws={"color": "red"}, ci=95)
r = sns.regplot(x="x", y="y4", data=data, scatter_kws={"s":5}, line_kws={"color": "green"}, )

labels = ['y^', 'y']

l._legend.texts[0].set_text(labels[0])
r._legend.texts[0].set_text(labels[1])

Здесь l и r возвращают экземпляр графика, а затем вы используете _legend дляполучить доступ к соответствующим легендам и set_text, чтобы переименовать текст по вашему выбору.