Python에서 파일 확장자를 추출하는 방법: os.path.splitext 활용법

Python에서는 os.path.splitext 메서드를 사용하면 파일 이름 문자열에서 확장자를 손쉽게 추출할 수 있습니다.

이 메서드는 경로(path)를 (root, ext) 형태의 튜플로 분리합니다. 이때 다음 조건이 성립합니다.

• root + ext == path: 분리된 두 요소를 합치면 원래 경로와 동일합니다.
• ext(확장자)는 빈 문자열이거나, 마침표(.)로 시작하며 최대 하나의 마침표만 포함합니다.

사용 예제

다음은 os.path.splitext를 사용하여 파일 확장자를 추출하는 기본 예제입니다.

import os

file_name = 'my_file.txt'
print(os.path.splitext(file_name))

실행 결과

위 코드를 실행하면 아래와 같은 출력을 확인할 수 있습니다.

('my_file', '.txt')

출력 결과를 보면 파일 이름인 'my_file'과 확장자인 '.txt'가 튜플 형태로 분리된 것을 알 수 있습니다. 확장자만 필요한 경우 인덱스를 사용하여 다음과 같이 추출할 수 있습니다.

import os

file_name = 'my_file.txt'
ext = os.path.splitext(file_name)[1]
print(ext)  # 결과: .txt

또한 확장자 앞의 마침표를 제거하고 싶다면 lstrip('.') 또는 슬라이싱([1:])을 활용하면 됩니다. 확장자가 없는 파일 이름의 경우 ext 값은 빈 문자열 ''로 반환되므로, 별도의 예외 처리 없이 안전하게 사용할 수 있다는 점도 장점입니다.