Python中所有子图标签Legend显示详解

在数据可视化中，图例（legend）是一个非常重要的元素，它能够帮助读者理解图表中不同元素的含义。特别是在使用Python进行可视化时，matplotlib库是一个非常强大的工具，能够轻松创建包含多个子图的图表，并在每个子图中显示图例。本文将详细介绍如何在Python的matplotlib库中为所有子图显示标签legend，包括理论概述和详细的代码示例。

一、理论概述

1.图例（Legend）的作用

• 图例用来解释绘图中各种元素的符号，帮助观众理解每种线条、颜色或符号代表的数据。例如，在一个折线图中，通过图例可以清晰地了解到每一条线代表的是哪个数据集。

2.matplotlib中的legend函数

• matplotlib.pyplot.legend(*args, **kwargs)：用于创建图例。
• loc参数：设置图例的位置，如'upper right'、'lower left'等。
• fontsize参数：设置图例的字体大小。
• frameon参数：设置是否显示图例边框。
• edgecolor和facecolor参数：分别设置图例边框和背景的颜色。
• title参数：设置图例的标题。

3.在多个子图中显示图例

• 使用plt.subplots()方法创建包含多个子图的图表。
• 每个子图可以单独调用legend()方法显示图例。
• 也可以使用fig.legend()方法在整个图形上方添加一个全局图例。

二、代码示例

以下是一个详细的代码示例，展示了如何在多个子图中显示图例。

import matplotlib.pyplot as plt
import numpy as np
 
# 生成数据
x = np.linspace(0, 10, 100)
y1 = np.sin(x)
y2 = np.cos(x)
 
# 创建包含两个子图的图表
fig, axs = plt.subplots(2)
 
# 在第一个子图中绘制 sin(x)
axs[0].plot(x, y1, label='sin(x)', color='blue')
axs[0].set_title('Sine Function')
axs[0].legend()  # 添加图例
 
# 在第二个子图中绘制 cos(x)
axs[1].plot(x, y2, label='cos(x)', color='orange')
axs[1].set_title('Cosine Function')
axs[1].legend()  # 添加图例
 
# 调整布局
plt.tight_layout()
plt.show()

在上述代码中，我们创建了一个包含两个子图的图表，每个子图都有自己的图例。通过label参数为每个数据系列指定标签，并在每个子图中调用legend()方法显示图例。

三、全局图例的显示

如果你想在整个图形上方添加一个全局图例，可以使用fig.legend()方法。以下是一个示例：

import matplotlib.pyplot as plt
import numpy as np
 
# 生成数据
x = np.linspace(0, 10, 100)
y1 = np.sin(x)
y2 = np.cos(x)
 
# 创建包含两个子图的图表
fig, axs = plt.subplots(2)
 
# 在第一个子图中绘制 sin(x)
axs[0].plot(x, y1, label='sin(x)', color='blue')
axs[0].set_title('Sine Function')
 
# 在第二个子图中绘制 cos(x)
axs[1].plot(x, y2, label='cos(x)', color='orange')
axs[1].set_title('Cosine Function')
 
# 在整体图中添加图例
fig.legend(loc='upper center', ncol=2)
 
# 调整布局
plt.tight_layout()
plt.show()

在这个示例中，我们使用fig.legend()方法在整个图形上方添加了一个全局图例，并且设置了图例的位置为'upper center'，列数为2。这样不仅保持了每个子图的独立性，同时也避免了重复内容。

四、图例的样式调整

除了设置图例的位置，还可以调整图例的样式，如字体大小、边框和背景颜色等。以下是一个示例：

import matplotlib.pyplot as plt
import numpy as np
 
# 生成数据
x = np.linspace(0, 10, 100)
y1 = np.sin(x)
y2 = np.cos(x)
 
# 创建包含两个子图的图表
fig, axs = plt.subplots(2)
 
# 在第一个子图中绘制 sin(x)
axs[0].plot(x, y1, label='sin(x)', color='blue')
axs[0].set_title('Sine Function')
 
# 设置图例样式
legend = axs[0].legend(loc='upper left', fontsize='x-large', frameon=False, edgecolor='blue', facecolor='lightgray')
 
# 在第二个子图中绘制 cos(x)
axs[1].plot(x, y2, label='cos(x)', color='orange')
axs[1].set_title('Cosine Function')
 
# 设置第二个子图的图例样式
legend2 = axs[1].legend(loc='upper right', fontsize='medium', frameon=True, edgecolor='red', facecolor='white')
 
# 调整布局
plt.tight_layout()
plt.show()

在这个示例中，我们分别为两个子图设置了不同的图例样式。第一个子图的图例没有边框，背景颜色为浅灰色，字体大小为x-large，边缘颜色为蓝色。第二个子图的图例有边框，背景颜色为白色，字体大小为medium，边缘颜色为红色。

五、图例位置的调整

有时候，我们可能需要将图例放置在图表之外的位置，这时可以使用bbox_to_anchor参数。以下是一个示例：

import matplotlib.pyplot as plt
import numpy as np
 
# 生成数据
x = np.linspace(0, 10, 100)
y1 = np.sin(x)
y2 = np.cos(x)
 
# 创建包含两个子图的图表
fig, axs = plt.subplots(2)
 
# 在第一个子图中绘制 sin(x)
axs[0].plot(x, y1, label='sin(x)', color='blue')
axs[0].set_title('Sine Function')
 
# 获取当前子图的位置
box = axs[0].get_position()
# 调整子图位置，为图例留出空间
axs[0].set_position([box.x0, box.y0, box.width, box.height * 0.8])
 
# 在图表外部添加图例
axs[0].legend(loc='center', bbox_to_anchor=(0.5, 1.2), ncol=2)
 
# 在第二个子图中绘制 cos(x)
axs[1].plot(x, y2, label='cos(x)', color='orange')
axs[1].set_title('Cosine Function')
 
# 调整布局
plt.tight_layout()
plt.show()

在这个示例中，我们首先获取了第一个子图的位置，然后调整了子图的高度，为图例留出空间。接着，使用bbox_to_anchor参数将图例放置在图表外部的中心位置。

六、结论

在数据可视化中，合理使用图例可以极大提升图表的可读性。在Python中，利用matplotlib创建的子图可以很容易地添加图例，无论是为每个子图单独添加，还是统一在一起。本文详细介绍了如何在多个子图中显示图例，包括全局图例的显示、图例样式的调整和图例位置的调整等。通过这些方法，你可以更灵活地创建具有丰富信息的图表，帮助观众更好地理解数据。