如何使用ax.legend()为条形图的子图添加通用图例

在使用Matplotlib绘制条形图的子图时，可以使用ax.legend()方法为子图添加通用图例。ax.legend()方法用于为图表添加图例，图例是用于解释图表中各个元素的标签。

下面是使用ax.legend()为条形图的子图添加通用图例的步骤：

1. 首先，导入所需的库和模块：

import matplotlib.pyplot as plt
import numpy as np

1. 创建一个包含子图的图表对象，并设置子图的布局：

fig, ax = plt.subplots(nrows=1, ncols=2, figsize=(10, 5))

这里创建了一个包含1行2列的子图布局，每个子图的大小为10x5。

1. 生成条形图的数据：

x = np.array([1, 2, 3])
y1 = np.array([4, 5, 6])
y2 = np.array([7, 8, 9])

这里使用NumPy库生成了两组条形图的数据。

1. 在第一个子图中绘制条形图：

ax[0].bar(x, y1, label='Bar 1')
ax[0].bar(x, y2, bottom=y1, label='Bar 2')

这里使用ax[0]表示第一个子图，使用bar()方法绘制条形图。通过设置bottom参数，可以将第二组条形图放置在第一组条形图的上方。

1. 在第二个子图中绘制条形图：

ax[1].bar(x, y1, label='Bar 1')
ax[1].bar(x, y2, bottom=y1, label='Bar 2')

这里使用ax[1]表示第二个子图，使用bar()方法绘制条形图。

1. 使用ax.legend()方法为子图添加通用图例：

ax[0].legend()

这里使用ax[0]表示第一个子图，调用legend()方法添加图例。

1. 显示图表：

plt.show()

这里调用show()方法显示图表。

通过以上步骤，就可以使用ax.legend()为条形图的子图添加通用图例。在图例中，可以显示每个条形图的标签，以便更好地解释图表中的数据。