[스크립팅 #1] Python으로 파일명기준 폴더만들어서 파일옮기기

 

cat.1.jpg

상기와 같이

분류라벨.숫자.파일확장자

 

이렇게 파일명의 규칙이 있어 sub폴더를 분류라벨명으로 만들고 사진 파일들을 분리하고자 아래와 같이 Scripting하여 처리하엿습니다. 손으로 해도 되겟지만.. 이게 가장 확실하겠지요.

 

 

import os

data_path='E:\\DogvsCat'
train_data_path=data_path+'\\train'
print(train_data_path)
for root, dirs, files in os.walk(train_data_path):
    for file in files:
       str_class_name = file.split('.')[0]
       des_path = data_path+'\\train'+'\\'+str_class_name
       if not os.path.exists(des_path):
          os.makedirs(des_path)
       os.rename(train_data_path +'\\' +file, des_path +'\\' +file)             
             

 

파일을 옮길때는 다른것도 많겠지만 os.rename을 사용했습니다.

os.walk는 파일이랑 폴더명들을 하부까지 탐색할때 유용하지요.

 

이상입니다. 

 

즐코 합시다.