plt.scatter散点图加legend
时间: 2023-11-09 19:56:54 浏览: 119
Matplotlib scatter绘制散点图的方法实现
您可以通过使用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)
通过这个代码,您将得到一个带有图例的由变量着色的散点图。
阅读全文