python 之利用plt.legend()添加图例

plt.legend() 是Matplotlib库中用于添加图例（legend）的函数。图例是用于标识图表中各个数据系列的标签，以便读者可以理解图表中的不同数据含义。通过 plt.legend()，您可以将图例添加到您创建的图表中，以提高图表的可读性。以下是 plt.legend() 的详细介绍：

plt.legend(loc='best', prop={'size': 12}, title='图例标题')

plt.legend() 可以接受以下参数：

• loc：指定图例的位置。可以是字符串或整数。常用的位置包括：

  • 'best'（默认值）：自动选择最佳位置。
  • 'upper right'：右上角。
  • 'upper left'：左上角。
  • 'lower right'：右下角。
  • 'lower left'：左下角。
  • 'right'：右侧。
  • 'center left'：左侧中央。
  • 'center right'：右侧中央。
  • 'lower center'：底部中央。
  • 'upper center'：顶部中央。
  • 其他位置参数和整数参数也可以用于精确指定位置。

• prop：是一个字典，用于设置图例文本的属性，如字体大小、颜色等。在prop字典中，您可以使用键值对来指定各种属性，例如 {'size': 12, 'color': 'red'} 可以设置图例文本的大小为12，颜色为红色。

• title：是一个可选参数，用于设置图例的标题。

要使用plt.legend()，首先需要在plt.plot() 中为每个数据系列添加标签（使用 label 参数），然后在创建图例之前调用 plt.legend()。以下是一个示例：

import matplotlib.pyplot as plt

# 示例数据
x = [1, 2, 3, 4, 5]
y1 = [10, 15, 13, 18, 16]
y2 = [8, 12, 9, 14, 11]

# 绘制数据，并添加标签
plt.plot(x, y1, label='数据系列1')
plt.plot(x, y2, label='数据系列2')
plt.plot(x, y3, label='数据系列3')

# 添加图例
plt.legend()

# 显示图表
plt.show()

在上述示例中，plt.legend() 将自动识别数据系列的标签，并在图表中添加图例。图例位于默认的“最佳位置”。您可以使用其他参数来自定义图例的位置、样式和属性。图例不仅可以用于线图，还可以用于其他类型的图表，如散点图、柱状图等。