[Python] 파일 확장자 가져오기 (get file extension)

Python Get File Extension

 

다양한 방법이 있지만, 제가 선호하는 순서대로 작성해보겠습니다.

1. pathlib

• python 3.4 이상에서 사용
• 하나의 파일 이름에 점(.)이 여러개라도 가져옴

import pathlib

my_path = 'hello/bryan.zip'

a = pathlib.Path(my_path).suffix
print(a)
# '.zip'

.tar.gz 같은 경우는?

import pathlib

my_path = 'hello/bryan.tar.gz'

a = pathlib.Path(my_path).suffixes
print(a)
# ['.tar', '.gz']

참고로 tar.gz 로 끝나는 파일인지 체크하려면
''.join(a).endswith('.tar.gz')

 

2. os

import os

filename, file_extension = os.path.splitext('hello/bryan.zip')

print(filename)
# hello/bryan

print(file_extension)
# .zip

 

참고

왼쪽 . 을 빼려면

a = '.zip'

print(a[1:])
# 'zip'

print(a.lstrip('.'))
# 'zip'

첫번째 방법은 index error 가 발생할 수 있습니다.

둘다 None 일 경우에도 오류가 발생하죠