文章目录

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() 将自动识别数据系列的标签,并在图表中添加图例。图例位于默认的“最佳位置”。您可以使用其他参数来自定义图例的位置、样式和属性。图例不仅可以用于线图,还可以用于其他类型的图表,如散点图、柱状图等。