更改 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 和1Yes替换。谢谢

Answer 1

您可以使用便捷方法 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')

Answer 2

您可以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()

这会给你

并且应该删除自动生成的图例，只留下自定义图例。