특정 폴더에 있는 특정 확장자의 파일 리스트를 가져오기 위해 os.listdir() 또는 glob.glob()을 사용할 수 있다.
1. os.listdir()
os.listdir()을 이용하여 탐색경로 하의 파일과 폴더명을 가져올 수 있다.
1) 1개 확장자 리스트 추출
1
2
3
4
5
| import os
file_list = os.listdir('./images')
img_files = [file for file in file_list if file.endswith('.jpg')]
|
1
| ['picture1.jpg', 'picture2.jpg']
|
2) 여러개 확장자 리스트 추출
1
2
3
4
5
| import os
file_list = os.listdir('./images')
img_files = [file for file in file_list if file.endswith(('.jpg', 'png'))]
|
1
| ['picture1.jpg', 'picture2.jpg', 'pic01.png', 'pic02.png']
|
2. glob.glob()
glob.glob()을 이용하여 탐색경로 하의 파일과 폴더명을 가져오는데 해당 탐색경로도 같이 추출된다.
1) 1개 확장자 리스트 추출
1
2
3
4
| import glob
img_files = glob.glob('./images/*.jpg')
|
1
| ['./images/picture1.jpg', './images/picture2.jpg']
|
2) 여러개 확장자 리스트 추출
1
2
3
4
5
6
| import glob
img_files = []
for ext in ('*.jpg', '*.png'):
img_files.extend(glob.glob(os.path.join('./images', ext)))
|
1
| ['./images/picture1.jpg', './images/picture2.jpg', './images/pic01.png', './images/pic02.png']
|