Matplotlib中多个子图使用一个图例

在Matplotlib中，图例是一个非常重要的元素，它可以帮助我们区分不同的数据系列。有时候，我们可能希望在一个图形中有多个子图，并且希望这些子图共享一个图例。以下是如何实现这一目标的方法：
首先，我们需要创建一个图形窗口，然后在其中添加子图。我们可以通过 subplots 或 subplot2grid 函数来创建子图。
假设我们有两个子图，每个子图都绘制了不同的数据系列。为了使这两个子图共享一个图例，我们需要创建一个图例对象，并将其添加到每个子图的 axes 对象中。
以下是一个简单的示例代码：

import matplotlib.pyplot as plt
import numpy as np
# 创建数据
x = np.linspace(0, 10, 100)
y1 = np.sin(x)
y2 = np.cos(x)
# 创建图形和子图
fig, ax1 = plt.subplots()
ax2 = plt.subplot2grid((3, 3), (0, 0, 2, 2), rowspan=3)
# 绘制第一个子图的数据系列并添加图例
line1, = ax1.plot(x, y1, 'r-')
legend1 = ax1.legend([line1], ['sin(x)'])
ax1.add_artist(legend1)
# 绘制第二个子图的数据系列并添加到共享的图例中
line2, = ax2.plot(x, y2, 'b-')
legend1.legendHandles.append(line2)  # 将第二个子图的数据系列添加到第一个子图的图例中
# 显示图形
plt.show()

在这个示例中，我们创建了一个图形窗口，并添加了两个子图。然后，我们在第一个子图中绘制了一个数据系列并创建了一个图例。接着，我们将第二个子图中绘制的数据系列添加到了第一个子图的图例中。这样，两个子图就共享了一个图例。
需要注意的是，这种方法只适用于两个子图的情况。如果有更多的子图需要共享一个图例，那么我们需要创建一个包含所有需要共享的图例的列表，并在每个子图中添加这个列表中的图例。此外，这种方法的一个潜在问题是，如果其中一个子图的图例太大或太小，那么它可能会遮挡住其他子图中的数据。因此，在创建共享的图例时，我们需要仔细考虑如何合理地设置其位置和大小。