Автор оригинала: Pankaj Kumar.
Мы можем использовать функцию модуля Python OS SPLITEXT (), чтобы получить расширение файла. Эта функция разделяет путь к файлу в кортеж, имеющую два значения – root и расширение.
Получение расширения файла в Python
Вот простая программа, чтобы получить расширение файла в 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.