如何用Matplotlib在单画布中展示主文件夹下各子文件夹的前N张图片？

嘿，我来帮你搞定这个需求！你已经能拿到每个子文件夹的前5张图片路径了，接下来只需要把这些图片按布局画到同一个Matplotlib画布上就行，我给你整理了完整的实现步骤和代码：

核心思路 #

• 先根据子文件夹数量和每个文件夹要展示的图片数量，创建对应的子图网格（比如3个子文件夹×5张图，就是3行5列的布局）
• 遍历每个子文件夹，依次读取前N张图片，放到对应的子图位置
• 最后调整子图间距，让整体布局更美观，再显示或保存画布

完整代码示例 #

import os
import matplotlib.pyplot as plt
import matplotlib.image as mpimg

# 主文件夹路径
main_folder = "main_folder"
# 每个子文件夹取前5张图
num_images_per_folder = 5

# 获取子文件夹列表（过滤掉非文件夹的内容，避免出错）
sub_folders = [f for f in os.listdir(main_folder) if os.path.isdir(os.path.join(main_folder, f))]
num_folders = len(sub_folders)

# 创建画布和子图：行数=子文件夹数，列数=每个文件夹取的图片数
fig, axes = plt.subplots(nrows=num_folders, ncols=num_images_per_folder, figsize=(15, 9))

# 遍历每个子文件夹
for folder_idx, folder_name in enumerate(sub_folders):
    current_folder_path = os.path.join(main_folder, folder_name)
    # 获取当前文件夹的前num_images_per_folder张图片（只取图片文件，这里假设是常见的jpg/png）
    image_files = [f for f in os.listdir(current_folder_path) if f.lower().endswith(('.png', '.jpg', '.jpeg'))][:num_images_per_folder]
    
    # 遍历当前文件夹的图片，放到对应子图
    for img_idx, img_file in enumerate(image_files):
        img_path = os.path.join(current_folder_path, img_file)
        img = mpimg.imread(img_path)
        # 获取当前子图的axes对象
        ax = axes[folder_idx, img_idx]
        ax.imshow(img)
        # 隐藏子图的坐标轴，让图片更整洁
        ax.axis('off')
        # 可以给子图加标题，比如文件夹名+图片名
        ax.set_title(f"{folder_name}\n{img_file}", fontsize=8)

# 调整子图之间的间距，避免重叠
plt.tight_layout()
# 显示画布
plt.show()
# 如果需要保存画布，取消下面的注释
# plt.savefig("combined_images.png", dpi=300, bbox_inches='tight')

关键细节说明 #

• 过滤非文件夹/非图片文件：新增了判断os.path.isdir确保只遍历子文件夹，同时过滤出图片格式的文件，避免读取非图片内容报错
• 子图布局：用plt.subplots一次性创建所有子图，通过索引folder_idx和img_idx定位每个图片的位置
• 美化调整：加上ax.axis('off')隐藏坐标轴，plt.tight_layout()自动调整间距，让整体布局更清爽
• 可扩展性：如果后续子文件夹数量或取图数量变化，只需要修改num_images_per_folder变量即可

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