<파이썬> 파일명, 경로 변경하기

다음은 파일 일괄처리에 도움이 되는 파이썬의 함수들이다.

os.listdir(path) # 파일목록전달
    os.chdir(path) # 작업하고 있는 디렉토리 변경
    os.getcwd() # 현재 프로세스의 작업 디렉토리 얻기
    os.path.abspath(filename) # 파일의 상대 경로를 절대 경로로 바꾸는 함수
    os.path.exists(filename) # 주어진 경로의 파일이 있는지 확인하는 함수
    os.curdir() # 현재 디렉토리 얻기
    os.pardir() # 부모 디렉토리 얻기
    os.sep() # 디렉토리 분리 문자 얻기
    os.path.basename(filename) # 파일명만 추출
    os.path.dirname(filename) # 디렉토리 경로 추출
    os.path.split(filename) # 경로와 파일명을 분리
    os.path.join(path,path) # \\패턴으로 경로를 만들어줌
    os.path.splitdrive(filename) # 드라이브명과 나머지 분리 (MS Windows의 경우)
    os.path.splitext(filename) # 확장자와 나머지 분리
    shutil.copy(path, newPath) # 파일 복사
    shutil.copy2(path, newPath) # 파일 복사 (설정값도 함께 복사)
    shutil.move(path, newPath) # 파일 이동줌
    

다음은 졸업과제를 진행하면서 CAFFE로 뉴럴네트워크를 학습시킬 때 데이터 전처리에 사용했던 파이썬 코드이다.

import os
    import shutil
    
    pivot = 22
    label_num = 14
    os.chdir('/home/nr-lab/Downloads/labeled_images/Label_' +str(label_num)+ '/')
    files = os.listdir('.')
    files.sort()
    fileNum = []
    for file in files:
        file = file[len('L'+str(label_num)+'_'):]
        file = file.rstrip('.jpg')
        fileNum.append(int(file))
    
    fileNum.sort()
    sortedFile = []
    for fn in fileNum:
        sortedFile.append('L'+str(label_num)+'_' + str(fn) + '.jpg')
    
    for name in sortedFile:
        if name.startswith('L'+str(label_num)+'_'+str(pivot)+'.jpg'):
            newname = 'L'+str(label_num-4)+'_'+str(count)+'_test.jpg'
            os.rename(name, newname)
            shutil.move('./'+newname, '../test/'+newname)
            count+=1
            pivot+=1