python 按文件名批量移动文件至指定文件夹

做猫狗分类任务,要求数据集划分为训练集(training dataset)和验证集(validation dataset),均包含dogs和cats两个目录,且每个目录下包含与目录名类别相同的RGB图。数据集共25000张照片,其中训练集猫狗照片各10000张,验证集猫狗照片各2500张。但我下载的数据只有一个包含所有猫狗图的文件夹,于是需要遍历所有文件并且移动,代码如下:

import os
import shutil

path = "C:\\Users\\Administrator\\Documents\\Python\\dnnlearn\\dogcat\\dataset"
src_dir = os.path.abspath(path + "\\train")
dst_dir = os.path.abspath(path + "\\dog")
cat_dir = os.path.abspath(path + "\\")
train__dir = os.path.abspath(src_dir + "\\train\\dog")

count = 0

if not os.path.exists(src_dir):#创建文件夹
    os.makedirs(src_dir)
#if not os.path.exists(train_cat_dir):
#    os.makedirs(train_cat_dir)

if os.path.exists(src_dir):
    for root, dirs, files in os.walk(src_dir):
        '''
        print("root:",root)#文件夹路径
        print("dirs:",dirs)#文件夹名称
        print("files:",files)#文件名
        '''
        for file in files:#遍历每一个文件
            #filename = os.path.basename(file)#获取文件名
            #if count <= 2500:
            dog_file = os.path.join(src_dir, file)#拼接得到文件路径
            shutil.move(dog_file, dst_dir)#移动文件
#print(count)

Python遍历文件好像是按照深度搜索进行的,有一个如下结构的文件夹:

python 按文件名批量移动文件至指定文件夹_第1张图片

遍历结果为:

python 按文件名批量移动文件至指定文件夹_第2张图片

 

你可能感兴趣的:(个人学习,python基础)