plt.scatter散点图加legend

您可以通过使用Matplotlib的legend()函数将图例添加到由变量着色的散点图中。在legend()函数中，您可以通过从情节中提取图例元素来指定标题和句柄。以下是一个示例代码：

plt.figure(figsize=(8,6))
scatter = plt.scatter(df.culmen_length_mm, df.culmen_depth_mm, s=150, c=df.species.astype('category').cat.codes)
plt.xlabel("Culmen Length", size=24)
plt.ylabel("Culmen Depth", size=24)

# 添加图例到情节中
plt.legend(handles=scatter.legend_elements(), title="species")

plt.savefig("scatterplot_colored_by_variable_legend.png", format='png',dpi=150)

通过这个代码，您将得到一个带有图例的由变量着色的散点图。