Извлечение расширения из имени файла в Python

Часто возникает ситуация, когда при работе с файлами нужно получить их расширение. Например, при обработке большого количества файлов разных типов, будет полезно знать, какой формат у каждого из них. Допустим, есть файл с именем «example.txt». Задача — получить расширение этого файла, то есть «txt».

Python предоставляет несколько способов для извлечения расширения из имени файла. Рассмотрим некоторые из них.

Использование модуля os

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

import os
filename = "example.txt"
basename, extension = os.path.splitext(filename)
print(extension)

В результате выполнения этого кода будет выведено расширение файла: «.txt». Отметим, что функция возвращает расширение вместе с точкой.

Использование модуля pathlib

Модуль pathlib является более современным и удобным способом работы с файловыми системами. Для извлечения расширения из имени файла можно использовать метод .suffix объекта Path.

from pathlib import Path
filename = "example.txt"
extension = Path(filename).suffix
print(extension)

Также, как и в случае с os.path.splitext(), расширение возвращается вместе с точкой.

Вывод

Python предоставляет несколько удобных методов для извлечения расширения из имени файла. Выбор метода зависит от конкретной ситуации и предпочтений разработчика.