Matplotlib add_subplot()在for循环中异常：2×3子图变为6×1

我编写了自定义函数create_subplot，用于从指定目录读取保存的图片并生成子图。预期生成2×3布局的子图，但实际输出为6×1布局。已确认循环计数器的迭代数值正确，恳请提供解决建议。

函数代码如下：

def create_subplot(my_direc,name_vec):
     #name_vec = 包含n个字符串的向量（nx1或1xn）
     #my_direc = 单个.png文件的存储路径

     counter = 1;
     for i in name_vec:
         genename = i.replace("/","_"); #替换基因名中的斜杠
         filname = my_direc + genename + '.png';
     
         fig = plt.figure()
         img = cv2.imread(filname); 
         img_RGB = cv2.cvtColor(img, cv2.COLOR_BGR2RGB); #将图像转换回原始颜色
     
         fig.add_subplot(2, 3, counter)
         plt.imshow(img_RGB)
         plt.axis('off'); plt.title("")
     
         print(counter)
         counter = counter + 1;

当前输出为6行1列的子图布局（每张图片单独占据一行）。

问题根源 #

循环内部每次都创建新的plt.figure()对象，导致每张图片都被分配到独立的画布中，最终呈现为6个单独的画布，看起来像6×1布局，而非同一个画布上的2×3子图。

修复方案 #

将画布创建语句fig = plt.figure()移到循环外部，确保所有子图都添加到同一个画布上：

def create_subplot(my_direc,name_vec):
     #name_vec = 包含n个字符串的向量（nx1或1xn）
     #my_direc = 单个.png文件的存储路径

     # 提前创建单个画布，放在循环外
     fig = plt.figure(figsize=(12, 8)) # 可根据需求调整画布尺寸
     counter = 1;
     for i in name_vec:
         genename = i.replace("/","_"); #替换基因名中的斜杠
         filname = my_direc + genename + '.png';
     
         img = cv2.imread(filname); 
         img_RGB = cv2.cvtColor(img, cv2.COLOR_BGR2RGB); #将图像转换回原始颜色
     
         # 向同一个画布添加子图
         ax = fig.add_subplot(2, 3, counter)
         ax.imshow(img_RGB)
         ax.axis('off'); ax.set_title("")
     
         print(counter)
         counter = counter + 1;
     # 自动调整子图间距，避免边缘截断
     plt.tight_layout()
     plt.show()

额外优化点 #

• 使用ax.set_title()替代plt.title()，避免依赖Matplotlib的隐式当前轴，代码更稳健
• 添加plt.tight_layout()自动优化子图间距，提升显示效果
• 通过figsize参数设置合适的画布大小，让2×3布局的图片比例更协调

内容的提问来源于stack exchange，提问作者Analia Quirk