使用 Matplotlib 创建带子图的图形并添加中文注释

在代码中，我已经创建了一个具有子图的图形，并设置了相应的标题和图像。现在，我将添加中文注释来解释代码的不同部分。

首先，我们通过使用plt.subplots函数来创建一个具有2行5列的子图，并指定了图形的大小。

fig, ax = plt.subplots(nrows=2, ncols=5, figsize=(15, 12))

接下来，我初始化了一个索引变量idx，用于迭代每个子图。

idx = 0

然后，我定义了一个中文字体用于标题的显示。你需要将'C:\Windows\Fonts\SimHei.ttf'替换为你自己的中文字体文件的路径。

font_prop = FontProperties(fname='C:\Windows\Fonts\SimHei.ttf')

通过使用两个嵌套的循环，我遍历了每个子图，并设置了相应的标题、图像和坐标轴。

for i in range(2):
    for j in range(5):
        label = labels[np.argmax(train_generator[0][1][idx])]
        ax[i, j].set_title(f"{label}", fontproperties=font_prop)  # 设置标题，使用中文字体
        ax[i, j].imshow(train_generator[0][0][idx][:, :, :])
        ax[i, j].axis("off")
        idx += 1

最后，我进行了一些布局调整，并给整个图形添加了一个标题。

plt.tight_layout()
plt.suptitle("训练样本图像示例\n", fontsize=21, fontproperties=font_prop)  # 设置标题，使用中文字体
plt.show()

通过运行代码，你将获得一个具有中文注释的图形，其中包含了训练样本图像的示例。