更改Seaborn散点图中的图例位置和标签

Question

我想在Seaborn散点图中更改位置以及我的传说的标签。这是我的代码：

ax_total_message_ratio=sns.scatterplot(x='total_messages', y='email_messages_ratio',hue='poi',data=df_new)
ax_total_message_ratio.set_title("Email Messages Ratio vs. Total Messages Across Poi",y=1.12,fontsize=20,fontweight='bold')
ax_total_message_ratio.set_ylabel('Email Messages Ratio')
ax_total_message_ratio.set_xlabel('Total Messages')
ax_total_message_ratio.legend.loc("lower right")
put.show()

但是我收到以下错误消息；    'function' object has no attribute 'loc'。我可以在Seaborn如何控制传奇方面获得一些帮助吗？此外，我还需要在图例标签中将0替换为No，将1替换为Yes。谢谢

Answer 1

您可以在调用loc时使用legend()关键字来调整位置，例如

ax_total_message_ratio.legend(loc="lower right")

要包括标记的自定义标签，您可以创建自定义图例，即

import matplotlib.pyplot as plt
from matplotlib.lines import Line2D

custom = [Line2D([], [], marker='.', color='b', linestyle='None'),
          Line2D([], [], marker='.', color='r', linestyle='None')]

fig = plt.figure(figsize=(8,5))

plt.legend(custom, ['Yes', 'No'], loc='lower right')
plt.show()

这会给你

和应该删除自动生成的图例，仅保留自定义图例。

Answer 2

您可以使用便捷方法 get_legend_handles_labels() 来获取标签的句柄和文本：

ax=sns.scatterplot(x='total_messages', y='email_messages_ratio',hue='poi',data=df_new)

handles, labels  =  ax.get_legend_handles_labels()

ax.legend(handles, ['Yes', 'No'], loc='lower right')