특정 파일 리스트 가져오기(listdir)

os.listdir

우선 os 모듈의 listdir을 사용해 특정 폴더에 있는 .py 파일만 보게 된다.

import os path = "./" file_list = os.listdir(path) print ("file_list: {}".format(file_list))

코드는 매우 간단하다.

현재 디렉토리에 있는 모든 파일(디렉토리) 리스트를 가져온다.

실행 결과는 다음과 같다.

file_list: ['a','b','c'];

디렉토리 내에 있는 모든 파일 및 디렉토리 리스트가 나왔다.

내가 원하는 파일은 .py 확장자를 가진 파일이므로, 코드를 추가해보면

import os path = "./" file_list = os.listdir(path) file_list_py = [file for file in file_list if file.endswith(".py")] print ("file_list_py: {}".format(file_list_py))

file_list에서 파일명이 .py로 끝나는 데이터만 file_list_py에 다시 정의했다.

 

file_list_py: ['etl.py', 'find_file.py', 'asd.py', 'namespace_.py', 'namespace_2.py', 'eq.py']

확장자가 py인 파일만 정상적으로 추출해냈다.