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

안녕하세요.

오늘은 파이썬에서 특정 파일 이름이나 형식을 리스트로 가져오는 방법에 대해 알아보겠습니다.

 

 파이썬 파일 리스트 가져오기

- 특정 폴더에 있는 특정 이름이나 파일 형식의 리스트를 가져와야 할 때, os 혹은 glob 모듈을 활용하면 됨.

 

▣ os 모듈 활용하기

import os

direct = "C:/Users/코딩하는 금융인/Desktop/배경화면"

file_list = os.listdir(direct)
jpg_list = []

# jpg 형식 파일 리스트 추출
for file in file_list:
    if file.endswith(".jpg"):
    	jpg_list.append(file)

print(jpg_list)

: 출력 결과

['44194284702_b7b0bb922a_b.jpg', '7690678692_85f4b41897.jpg', 'CI-ARS-USDA-Greb-013-600-400px.jpg', 'img.jpg', 'money-finance-cash-coins.jpg', 'network-sustainability-generic-newjpg_50677.jpg', 'refugees-economic-migrants-financial-equalization-help.jpg', '다운로드 (1).jpg', '다운로드 (2).jpg', '다운로드 (3).jpg', '다운로드.jpg', '록펠러뷰.jpg', '록펠러뷰2 .jpg', '록펠러뷰3.jpg', '센트럴파크.jpg', '센트럴파크2ty.jpg', '썸네일.jpg']

 

▣ glob 모듈 활용하기

import glob

input_path = "C:/Users/코딩하는 금융인/Desktop/배경화면/*.jpg"
file_list = [file for file in glob.glob(input_path)]

print(file_list)

: 출력 결과

['C:/Users/코딩하는 금융인/Desktop/배경화면\\44194284702_b7b0bb922a_b.jpg', 'C:/Users/코딩하는 금융인/Desktop/배경화면\\7690678692_85f4b41897.jpg', 'C:/Users/코딩하는 금융인/Desktop/배경화면\\CI-ARS-USDA-Greb-013-600-400px.jpg', 'C:/Users/코딩하는 금융인/Desktop/배경화면\\img.jpg', 'C:/Users/코딩하는 금융인/Desktop/배경화면\\money-finance-cash-coins.jpg', 'C:/Users/코딩하는 금융인/Desktop/배경화면\\network-sustainability-generic-newjpg_50677.jpg', 'C:/Users/코딩하는 금융인/Desktop/배경화면\\refugees-economic-migrants-financial-equalization-help.jpg', 'C:/Users/코딩하는 금융인/Desktop/배경화면\\다운로드 (1).jpg', 'C:/Users/코딩하는 금융인/Desktop/배경화면\\다운로드 (2).jpg', 'C:/Users/코딩하는 금융인/Desktop/배경화면\\다운로드 (3).jpg', 'C:/Users/코딩하는 금융인/Desktop/배경화면\\다운로드.jpg', 'C:/Users/코딩하는 금융인/Desktop/배경화면\\록펠러뷰.jpg', 'C:/Users/코딩하는 금융인/Desktop/배경화면\\록펠러뷰2 .jpg', 'C:/Users/코딩하는 금융인/Desktop/배경화면\\록펠러뷰3.jpg', 'C:/Users/코딩하는 금융인/Desktop/배경화면\\센트럴파크.jpg', 'C:/Users/코딩하는 금융인/Desktop/배경화면\\센트럴파크2ty.jpg', 'C:/Users/코딩하는 금융인/Desktop/배경화면\\썸네일.jpg']

 

▣ 결과 비교하기

- os 모듈로 파일 리스트를 가져오면 해당 디렉토리의 파일명만 가져왔지만, glob 모듈로 파일 리스트를 가져오는 경우에는 경로까지 포함한 파일명을 가져옴.

- 어떤 상황이냐에 따라 다르게 활용할 수 있으므로 두 가지 모듈을 상황에 맞게 활용하는 게 좋음.