Как получить расширение файла в Python

Мы можем использовать функцию splitext() модуля ОС Python, чтобы получить расширение файла. Эта функция разбивает путь к файлу на кортеж, имеющий два значения — корень и расширение.

Получение расширения файла в 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.