Python中的 Matplotlib.pyplot.legend()

Matplotlib 是 Python 中常用的绘图库，pyplot 子包是 Matplotlib 的部分，可以实现数据可视化。plt.legend() 函数用于为绘图添加图例。图例是解释图表内容的一种方式，在数据可视化中起到了非常重要的作用。

函数原型

legend(handles=None, labels=None, loc=None, bbox_to_anchor=None, shadow=False, frameon=False, framealpha=None, facecolor=None, edgecolor=None, title=None, title_fontsize=None, prop=None, labelspacing=None)

参数说明

• handles: 要显示图例的对象，如果没有此参数，则默认显示由 plt.plot() 创建的线条，如果有此参数，则将指定对象传递给它。本参数可以是线条对象、散点图对象、长条图等 Matplotlib 对象等。
• labels: 与 handles 对象匹配的文本标签列表。
• loc: 图例的位置
• bbox_to_anchor: 用户可以在此处定义图例框的位置。
• shadow: 是否添加阴影效果。
• frameon: 是否添加边框。
• framealpha: 边框透明度。
• facecolor: 图例的颜色。
• edgecolor: 图例边框的颜色。
• title: 图例的标签。
• title_fontsize: 图例标签的字体大小。
• prop: 文本属性。
• labelspacing: 标签间距。

代码示例

import numpy as np

import matplotlib.pyplot as plt

x = np.arange(1, 11)
y1 = x
y2 = x + 2
plt.plot(x, y1, label='y = x')
plt.plot(x, y2, label='y = x + 2')
plt.legend()

plt.show()

上述代码中，我们使用 plt.plot() 函数绘制了两条曲线，使用 plt.legend() 函数为曲线添加图例。函数默认将 plt.plot() 函数绘制的两条曲线添加至图例中，并显示在最佳位置，用户也可以使用 loc 参数自定义图例的位置、使用 handles 和 labels 参数自定义图例中的展示样式。