[Python] 폴더 내 존재하는 파일을 텍스트 파일에 쓰기

os 모듈을 사용하여 지정된 폴더내의 모든 파일명을 가져와서 해당 파일명들을 텍스트 파일에 저장하는 간단한 파이썬 코드입니다.

 

데이터 처리나 분석 과정에서 폴더 내 파일 리스트를 확인하고, 이를 텍스트 파일로 저장해야 할 경우에 유용하게 사용될 수 있습니다.

import os
from tqdm import tqdm

path = 'D:/cimjuk_final/test/images'
txt_file = 'D:/cimjuk_final/test/img_list.txt'
files = os.listdir(path)

for i in tqdm(files):
    f = open(txt_file, 'a')
    f.write(i+'\n')

print('done')

 

위 코드는 지정된 path 경로 내의 모든 파일 리스트를 가져옵니다.

 

그리고 tqdm 패키지를 사용하여 파일 리스트를 반복문으로 순회하면서 텍스트 파일에 파일명을 저장합니다.

 

파일은 open() 함수를 사용하여 텍스트 파일을 열고, write() 함수를 사용하여 파일명을 작성하고

 

마지막으로, 파일 쓰기 작업이 끝나면 print() 함수를 사용하여 "done" 메시지를 출력합니다.

 

위 코드를 실행해보면, 해당 경로 내의 모든 파일명이 텍스트 파일에 저장되는 것을 확인할 수 있고, tqdm 패키지를 사용하여 진행 상황을 표시하므로 처리 속도나 진행 상황을 확인할 수 있어서 좀 더 직관적이며

 

기다리기 싫어하는 한국인에게 안성맞춤입니다. (남은 시간 알려줌)

 

100%|██████████| 10/10 [00:00<00:00, 620.58it/s]
done

 

실행 결과