plt.subplot()调用问题

Question

调用plt.subplot()函数后图像单独显示，而不是所有图像在同一张大图上，该如何解决这个问题
代码如下：

plt.figure(figsize=(16,8))
for i in range(len(image[:6])):
    img=image[:6][i].numpy()
    img=np.squeeze(img)
    label_img=label[:6][i].numpy()
    plt.subplot(2,4,i+1)
    plt.title(label_img)
    plt.imshow(img)
    plt.show()

输出如图

Answer 1

【以下回答由 GPT 生成】

问题分析： 从代码中可以看出，使用了plt.subplot()函数来创建子图，但是最终显示的子图总是不正确。希望得到正确的子图显示结果。

解决方案: plt.subplot()函数的三个参数分别表示子图布局的行数，列数和当前子图的索引。在代码中，设置的行数为2，列数为4，但是最终显示的子图数量超过了设定的范围。

修改代码的部分如下：

plt.figure(figsize=(16, 8))
for i in range(len(image[:6])):
    img = image[:6][i].numpy()
    img = np.squeeze(img)
    label_img = label[:6][i].numpy()
    plt.subplot(2, 3, i+1)  # 修改此处的列数为3
    plt.title(label_img)
    plt.imshow(img)
plt.show()  # 将plt.show()放在for循环外，只显示一次图像

如此一来，每次循环时，会自动在指定位置上创建子图，并进行图像显示。一共需要创建2行3列共6个子图来显示图片。

这样，就能够正确显示多个子图的结果了。

请注意，根据图片输出效果，需要根据具体需求调整子图的布局参数。

【相关推荐】

• 你看下这篇博客吧, 应该有用👉 ：plt.subplot()使用示例
• 除此之外, 这篇博客: 卷积神经网络中的 plt.subplot() 部分也许能够解决你的问题。

如果你已经解决了该问题, 非常希望你能够分享一下解决方案, 写成博客, 将相关链接放在评论区, 以帮助更多的人 ^-^