Python으로 파일명을 효율적으로 가져오는 방법 | os 모듈과 pathlib 모듈 완전 분석

1. 시작하기

2. Python에서 파일명 가져오기 방법의 기본

2.1 os.path.basename() 사용법

import os

file_path = "/user/home/document/file.txt"
file_name = os.path.basename(file_path)
print(file_name)  # 결과: "file.txt"

2.2 os.path.dirname()로 디렉터리명 가져오기

dir_name = os.path.dirname(file_path)
print(dir_name)  # 결과: "/user/home/document"

3. 디렉터리 내의 모든 파일 이름을 가져오기

3.1 os.listdir()를 사용한 파일 이름 가져오기

import os

dir_path = "/user/home/document"
files = os.listdir(dir_path)
print(files)  # 결과: ["file1.txt", "file2.txt", "subfolder"]

3.2 os.walk()로 재귀적으로 파일 이름을 가져오기

import os

dir_path = "/user/home/document"
for root, dirs, files in os.walk(dir_path):
    for file in files:
        print(os.path.join(root, file))

3.3 오류 처리 추가

dir_path = "/invalid/path"
try:
    files = os.listdir(dir_path)
    print(files)
except FileNotFoundError:
    print(f"{dir_path}를 찾을 수 없습니다.")

4. 확장자 가져오기 및 변경

4.1 os.path.splitext()로 확장자 가져오기

import os

file_path = "/user/home/document/file.txt"
file_name, file_ext = os.path.splitext(file_path)
print(file_name)  # 결과: "/user/home/document/file"
print(file_ext)   # 결과: ".txt"

4.2 확장자 변경하기

new_file_path = file_name + ".jpg"
print(new_file_path)  # 결과: "/user/home/document/file.jpg"

5. pathlib 모듈을 사용한 파일명 가져오기 응용

5.1 Path.iterdir()를 사용한 파일명 가져오기

from pathlib import Path

dir_path = Path("/user/home/document")
for item in dir_path.iterdir():
    print(item.name)

5.2 재귀적인 파일 가져오기: rglob()

for file in dir_path.rglob("*"):
    print(file)

6. 파일명 필터링

6.1 확장자로 필터링

from pathlib import Path

dir_path = Path("/user/home/document")
for file in dir_path.glob("*.txt"):
    print(file)

7. 요약