如何根据DataFrame的聚类结果按图片编号归类文件夹中的图片？

完全懂你的需求——把聚类好的图片按所属类别批量移到对应子文件夹，用pandas+shutil就能轻松完成，我给你整理了一套可直接复用的方案，还加了实用细节避坑：

具体步骤&代码 #

1. 先导入必备工具库 #

用pandas操作你的DataFrame，shutil负责文件移动，os处理路径和文件夹创建：

import pandas as pd
import shutil
import os

2. 提前创建所有聚类子文件夹 #

避免移动文件时因文件夹不存在报错，先一次性建好所有需要的子文件夹：

# 假设你的DataFrame名为df
unique_clusters = df['cluster'].unique()
# 替换成你的图片根文件夹的绝对路径
root_image_dir = "/path/to/your/image/folder"

for cluster_id in unique_clusters:
    cluster_folder_path = os.path.join(root_image_dir, f"cluster{cluster_id}")
    # exist_ok=True 保证文件夹已存在时不会抛出错误
    os.makedirs(cluster_folder_path, exist_ok=True)

3. 批量移动图片到对应聚类文件夹 #

遍历DataFrame的每一行，根据cluster列的值把图片移到目标文件夹：

for _, row in df.iterrows():
    # 处理图片文件名：如果你的img列不带扩展名，记得加上（比如.jpg/.png，根据实际格式修改）
    img_filename = f"{row['img']}.jpg"  # 若img列已包含扩展名，直接用 row['img'] 即可
    # 拼接源文件和目标文件的完整路径
    source_file = os.path.join(root_image_dir, img_filename)
    target_file = os.path.join(root_image_dir, f"cluster{row['cluster']}", img_filename)

    # 先检查源文件是否存在，避免报错
    if os.path.exists(source_file):
        shutil.move(source_file, target_file)
        print(f"已完成移动：{img_filename} → cluster{row['cluster']}")
    else:
        print(f"警告：文件不存在，已跳过：{source_file}")

几个实用提醒 #

• 路径要准确：务必把root_image_dir替换成你实际的图片文件夹绝对路径，不然会找不到文件
• 先备份原文件：建议先复制一份图片文件夹做备份，避免移动过程中误操作导致文件丢失
• 扩展名适配：如果你的图片是.png/.jpeg等其他格式，记得修改代码里的扩展名部分
• 性能不用担心：868张图片的规模用iterrows()完全没问题，要是以后处理上万级别的数据，再换成df.apply()优化效率就行

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