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

특정 폴더에 있는 특정 확장자의 파일 리스트를 가져오기 위해 os.listdir() 또는 glob.glob()을 사용할 수 있다.

1. os.listdir()

os.listdir()을 이용하여 탐색경로 하의 파일과 폴더명을 가져올 수 있다.

1) 1개 확장자 리스트 추출

import os


file_list = os.listdir('./images')
img_files = [file for file in file_list if file.endswith('.jpg')]

------------------------------------------------------------
['picture1.jpg', 'picture2.jpg']

2) 여러개 확장자 리스트 추출

import os


file_list = os.listdir('./images')
img_files = [file for file in file_list if file.endswith(('.jpg', 'png'))]

------------------------------------------------------------
['picture1.jpg', 'picture2.jpg', 'pic01.png', 'pic02.png']

2. glob.glob()

glob.glob()을 이용하여 탐색경로 하의 파일과 폴더명을 가져오는데 해당 탐색경로도 같이 추출된다.

1) 1개 확장자 리스트 추출

import glob


img_files = glob.glob('./images/*.jpg')

------------------------------------------------------------
['./images/picture1.jpg', './images/picture2.jpg']

2) 여러개 확장자 리스트 추출

import glob


img_files = []
for ext in ('*.jpg', '*.png'):
  img_files.extend(glob.glob(os.path.join('./images', ext)))

------------------------------------------------------------
['./images/picture1.jpg', './images/picture2.jpg', './images/pic01.png', './images/pic02.png']