파이썬 | 폴더에 있는 파일 리스트 불러오기 (with. os 라이브러리)

이 글에서는 파이썬 코드를 이용하여 폴더에 있는 파일을 불러오는 방법에 대해 정리하고자 한다.

 

파이썬 코드로 데이터 정리 혹은 분석을 하다보면 여러 파일을 불러와야 할 때가 있는데, 이때 유용하게 사용할 수 있다.

 

코드를 실행한 환경은 다음과 같다.

... Windows 10 Home 22H2
... Python ver. 3.12.2
... Visual Studio Code
... Jupyter Notebook

---

특정 폴더 내 파일 리스트 불러오기

파이썬에서 폴더에 있는 파일을 리스트로 가져올 때에는 os.listdir 함수를 사용하면 된다. os.listdir 함수를 사용하기 위해서는 os 라이브러리를 import 해야한다.

import os

path = "경로를 입력하세요."
os.listdir(path)

폴더 경로에 있는 파일을 리스트로 반환

os.listdir 함수를 사용하면 경로에 있는 파일을 리스트로 반환한다.

 

for문을 사용하여면 파일을 하나씩 반복 작업을 수행할 수 있다.

for file in os.listdir(path):
    # 반복 작업 수행 코드 작성
    print(file)

 

if문 혹은 정규식을 이용하면 특정 문자열이 포함된 파일만 분리하는 것이 가능하다.

for file in os.listdir(path):
    if 'TXT' in file:
        # 수행 코드 작성
        print(file)

파일명에 TXT 문자열을 포함한 파일만 출력

---

요약

• 특정 폴더에 있는 파일을 리스트로 반환이 필요할 때에는 os.listdir 함수 사용
• 불러온 파일을 개별로 컨트롤 할 때에는 if문과 for문 사용