[파이썬, Python] 디렉토리(폴더)인지 파일인지 구분

파이썬 디렉토리(폴더)인지 파일인지 구분

만일 폴더 안의 구조가 아래와 같이 되어 있다고 하자. 여기서 폴더는 1학년 1반, 1학년 2반, 1학년 3반이고 파일은 README.md, test.txt, text.txt이다.

 

파이썬을 통해 폴더인지 확인하고 싶다면 os.path.isdir()함수를 사용하면 된다. 파일인지 확인하고 싶다면 os.path.isfile()함수를 사용하면 된다. 아래 예시 코드를 참고하면 좋을 것 같다.

import os
# 폴더 안에 있는 모든 폴더와 파일을 list에 저장
list = os.listdir('폴더 경로')
# ['1학년 1반', '1학년 2반', '1학년 3반', 'README.md', 'test.txt', 'text.txt']


# os.path.isdir()함수를 사용해서 폴더인지 확인
folders = [x for x in list if os.path.isdir(x)]
# ['1학년 1반', '1학년 2반', '1학년 3반']


# os.path.isfile()함수를 사용해서 파일인지 확인
files = [x for x in list if os.path.isfile(x)]
# ['README.md', 'test.txt', 'text.txt']