如何在Google Colab中将dataframe内的图像文件拆分到不同文件夹？

• 第一步：挂载Google Drive（如果数据集已直接上传到Colab运行实例中可跳过此步）

from google.colab import drive
drive.mount('/content/drive')

• 第二步：导入所需依赖库

import pandas as pd
import os
import shutil

• 第三步：创建存放分类结果的目标文件夹

# 可自行修改为你需要的文件夹存储路径
save_root = "/content/covid_dataset_split"
os.makedirs(os.path.join(save_root, "0"), exist_ok=True)
os.makedirs(os.path.join(save_root, "1"), exist_ok=True)

• 第四步：遍历DataFrame执行拆分操作

请将代码中的列名替换为你自己DataFrame对应的列名

• 如果你的DataFrame中存储的是图像本地路径：

# 假设你的DataFrame变量名为df，图像路径列名为image_path，标签列名为label
for idx, row in df.iterrows():
    img_path = row["image_path"]
    label = str(row["label"])
    target_path = os.path.join(save_root, label, os.path.basename(img_path))
    # 复制图像到目标文件夹，如需移动原文件可替换为shutil.move
    shutil.copy(img_path, target_path)

• 如果你的DataFrame中存储的是图像像素矩阵：

from PIL import Image
import numpy as np

# 假设你的DataFrame变量名为df，像素矩阵列名为img_matrix，标签列名为label
for idx, row in df.iterrows():
    img_matrix = row["img_matrix"]
    label = str(row["label"])
    target_path = os.path.join(save_root, label, f"{idx}.png")
    Image.fromarray(np.uint8(img_matrix)).save(target_path)

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