Matplotlib图例(legend)的设置与自定义

2024-01-03 12:38:08

Matplotlib 图例：让您的图表更具说明性

导言

在数据可视化的世界中，Matplotlib 是一个强大的库，可帮助您创建引人入胜且信息丰富的图表。为了进一步增强您的图表，Matplotlib 提供了一个方便的功能，称为图例。图例充当图表中不同数据系列的指南，使用户能够轻松识别和理解它们。

Matplotlib 图例

图例是一个矩形框，通常位于图表右上角，但可以通过调整其位置来更有效地利用图表空间。它包含一个标记和标签的集合，每个标记和标签对应图表中的一个数据系列。

图例位置设置

在 Matplotlib 中，您可以使用 legend() 函数指定图例的位置。这个函数接受一个 loc 参数，它允许您从几个预定义的位置中进行选择，例如：

'best': 图例的最佳位置由 Matplotlib 自动确定。
'upper left': 图例放置在图表的左上角。
'upper right': 图例放置在图表的右上角。
'lower left': 图例放置在图表的左下角。
'lower right': 图例放置在图表的右下角。

图例大小设置

除了位置之外，您还可以控制图例的大小。 legend() 函数接受一个 bbox_to_anchor 参数，允许您设置图例的锚点和大小。锚点是指图例相对于图表的定位，而大小表示图例占据图表的百分比。

图例样式设置

除了位置和大小之外，您还可以自定义图例的样式，包括边框、透明度、阴影和标题。 legend() 函数提供以下参数来控制这些方面：

frameon: 控制边框的显示（True/False）。
framealpha: 设置边框的透明度（0-1）。
shadow: 为图例添加阴影（True/False）。
title: 设置图例的标题。

代码示例

以下 Python 代码展示了如何使用 Matplotlib 创建和自定义图例：

import matplotlib.pyplot as plt

# 创建数据
x = [1, 2, 3, 4, 5]
y1 = [2, 4, 6, 8, 10]
y2 = [1, 3, 5, 7, 9]

# 创建图表
plt.plot(x, y1, label="数据系列 1")
plt.plot(x, y2, label="数据系列 2")

# 设置图例位置
plt.legend(loc='upper left')

# 设置图例大小
plt.legend(bbox_to_anchor=(1, 1))

# 设置图例样式
plt.legend(frameon=False, framealpha=0.5, shadow=True, title='图例')

# 显示图表
plt.show()