根据csv文件按照标签划分文件夹数据集

最新推荐文章于 2023-07-06 17:01:48 发布

原创最新推荐文章于 2023-07-06 17:01:48 发布

· 1.2k 阅读

2 ·

版权

文章标签：

#机器学习 #深度学习 #分类

该脚本读取csv文件，遍历每一项，检查目标路径是否存在，如果不存在则创建，并将原始路径的文件移动到目标文件夹下按类别进行分类。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

import csv
import shutil
import os

target_path = '.\\'#目标文件夹
original_path = '.\\train\\train\\'#原始文件夹
with open('.\\train\\train_sorted.csv',"rt", encoding="utf-8") as csvfile:
    reader = csv.reader(csvfile)
    rows= [row for row in reader]
    for row in rows:
        if os.path.exists(target_path+row[1]) :
            full_path = original_path + row[0]
            shutil.move(full_path,target_path + row[1] +'/')
        else :
            os.makedirs(target_path+row[1])
            full_path = original_path + row[0]
            shutil.move(full_path,target_path + row[1] +'/')