이 글에서는 파이썬 코드를 이용하여 폴더에 있는 파일을 불러오는 방법에 대해 정리하고자 한다.
파이썬 코드로 데이터 정리 혹은 분석을 하다보면 여러 파일을 불러와야 할 때가 있는데, 이때 유용하게 사용할 수 있다.
코드를 실행한 환경은 다음과 같다.
... 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)
요약
- 특정 폴더에 있는 파일을 리스트로 반환이 필요할 때에는
os.listdir
함수 사용 - 불러온 파일을 개별로 컨트롤 할 때에는
if
문과for
문 사용
'Development > Python' 카테고리의 다른 글
파이썬 | playwright | playwright로 웹 스크래핑 시 headless 여부에 따라 데이터를 받아오지 못할 때 해결 방법 (0) | 2024.05.19 |
---|