Часто возникает ситуация, когда при работе с файлами нужно получить их расширение. Например, при обработке большого количества файлов разных типов, будет полезно знать, какой формат у каждого из них. Допустим, есть файл с именем «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 предоставляет несколько удобных методов для извлечения расширения из имени файла. Выбор метода зависит от конкретной ситуации и предпочтений разработчика.
Добавить комментарий