创建一个包含 10 张训练图像的 2 行 5 列的子图

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

索引号初始化

idx = 0

使用两层循环遍历所有的单元格

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}') ax[i, j].imshow(train_generator[0][0][idx][:, :, :]) ax[i, j].axis('off') # 更新索引号 idx += 1

调整子图之间的间距，添加总标题，并将图表显示出来

plt.tight_layout() plt.suptitle('训练样本图像示例', fontsize=21) plt.show()