[Python] 디렉토리의 파일 리스트 가져오기 (os, glob)

오늘은 파이썬 코드에서 특정 디렉토리에 있는 파일 리스트를 가져오는 방법에 대해서 알아보겠습니다.

os 나 glob 모듈을 사용해서 가져올 수 있습니다.

두 가지의 결과가 살짝 다른데, 필요한 내용에 맞게 사용하시면 될 것 같아요.

 

1. os.listdir

이건 제가 주로 사용하는 코드입니다. 저는 먼가 os가 더 편하더라구요ㅎㅎ

 

import os

path = './'                   ## 확인하고 싶은 폴더의 경로를 작성합니다.
file_list = os.listdir(path)  ## 파일명만 저장됩니다.

## 특정 파일만 가져오고 싶을 때
## for문을 사용해서 특정 문자열을 포함한 것들만 다른 리스트에 저장하면 됩니다.
py_file_list = [f for f in file_list if '.py' in f]

 

2. glob

import glob

path      = './*'            ## 확인하고 싶은 폴더의 경로를 작성합니다.
file_list = glob.glob(path)  ## 파일명과 경로가 함께 저장됩니다.

## 특정 파일만 가져오고 싶을 때
## os를 import했을 때와 동일하게,
## for문을 사용해서 특정 문자열을 포함한 것들만 다른 리스트에 저장하면 됩니다.
py_file_list = [f for f in file_list if '.py' in f]