[파이썬] 하위경로의 파일 리스트 만들기

안녕하세요? 만들오 입니다.

 

하위경로의 파일 리스트는 두가지 방법으로 만들 수 있습니다.

 

1. 선택한 폴더를 대상으로 검색

import glob

def dir(start, ext):
    path = glob.glob('{}/*.{}'.format(start, ext))
    path = [filename.replace('\\', '/') for filename in path]
    return path
    
#example
path = dir('images\circle', 'png')

glob 모듈을 이용해서 선택한 폴더 내의 특정 확장자로된 파일을 검색하여 리스트로 반환합니다.

 

2. 선택한 폴더와 안에있는 폴더까지 모두 찾아내는 방식입니다.

import os

def dir_sub(start, ext):
    list = []
    for (path, dir, files) in os.walk(start):
        for filename in files:
            file_ext = os.path.splitext(filename)[-1]
            if ext == file_ext[1:]: # .png => png
                fullPath = "{}\{}".format(path, filename).replace('\\', '/')
                list.append(fullPath)
    return list
    #example
    path = dir('images', 'png')

* 이 글은 티스토리 카카오계정 연동정책으로 인해 이전 블로그(오코취) 글을 옮겨왔습니다.

[끝].