파이썬에서 파일 확장자를 얻는 방법

Python os 모듈 splitext() 함수를 사용하여 파일 확장자를 얻을 수 있습니다. 이 함수는 파일 경로를 루트와 확장자의 두 값을 갖는 튜플로 분할합니다.

파이썬에서 파일 확장자 얻기

다음은 Python에서 파일 확장자를 얻는 간단한 프로그램입니다.

import os

# unpacking the tuple
file_name, file_extension = os.path.splitext("/Users/pankaj/abc.txt")

print(file_name)
print(file_extension)

print(os.path.splitext("/Users/pankaj/.bashrc"))
print(os.path.splitext("/Users/pankaj/a.b/image.png"))

산출:

• 첫 번째 예에서는 튜플 값을 두 변수로 직접 압축 해제합니다.
• .bashrc 파일에는 확장자가 없습니다. 숨김 파일로 만들기 위해 파일 이름에 점이 추가됩니다.
• 세 번째 예에서는 디렉토리 이름에 점이 있습니다.

Pathlib 모듈을 사용하여 파일 확장자 가져오기

파일 확장자를 얻기 위해 pathlib 모듈을 사용할 수도 있습니다. 이 모듈은 Python 3.4 릴리스에서 도입되었습니다.

>>> import pathlib
>>> pathlib.Path("/Users/pankaj/abc.txt").suffix
'.txt'
>>> pathlib.Path("/Users/pankaj/.bashrc").suffix
''
>>> pathlib.Path("/Users/pankaj/.bashrc")
PosixPath('/Users/pankaj/.bashrc')
>>> pathlib.Path("/Users/pankaj/a.b/abc.jpg").suffix
'.jpg'
>>> 

결론

파일 확장자를 가져오려면 항상 표준 방법을 사용하는 것이 좋습니다. 이미 os 모듈을 사용하고 있다면 splitext() 메서드를 사용하십시오. 객체 지향 접근 방식의 경우 pathlib 모듈을 사용하십시오.