[Python]폴더 내 파일 정보 파악하기 (os.walk())

프로젝트를 진행하면서 data를 구글 드라이브에 넣어 사용하다보니 우분투처럼 파일을 이동하거나 복사하려면 여간 힘든게 아니다...(사랑해요 우분투..)

그래서 구글 코랩을 이용해 파일을 처리하는데 꿀 정보를 올린다...

(os.가 괜찮은게 진짜 많은 듯. 필수로 공부해야겠다)

path = ~~~
for (root, directories, files) in os.walk(path):
	for dir in directories:
		dir_path = os.path.join(root, dir)
    #이게 진짜 꿀인듯... 
    for file in files:
###### 나같은 경우엔 이렇게 해서 파일 확장자를 변경하면서 이동함 ######
        file_name = file.split('.')
        file_name = file_name[0]
        file_path = os.path.join(root, file)
        destination = f'/content/drive/MyDrive/SIA_data/train/Total/image_png/{file_name}.png'
        try:
            shutil.copyfile(file_path, destination)
            print(f'{file_name}완료')
        except:
            fail_list.append(file_name)
            print(f'{file_name}안됨')
    print('==================== png 변형 끝 ====================')
        

이런 식으로 해당 폴더에 있는 파일명과 파일 경로를 파악할 수 있다.

• root: 탐색 중인 상위 폴더의 경로
• directories: 탐색 중인 경로의 폴더 이름들
• files: 탐색 중인 경로의 파일 이름들

• 참고 블로그