В мире программирования на Python существует множество способов решения самых разных задач. Одной из распространенных задач является составление списка файлов в каталоге, имеющих определенное расширение. В этой статье мы погрузимся в захватывающую область обработки файлов и рассмотрим несколько методов достижения этой цели. Итак, пристегните ремни и приготовьтесь раскрыть всю мощь Python!

Метод 1: использование os.listdir() и манипуляция строками
Модуль os в Python предоставляет удобную функцию os.listdir(), которая возвращает список всех файлов и каталогов в заданном каталоге. Мы можем объединить эту функцию с методами манипулирования строками для фильтрации файлов с определенным расширением. Вот пример фрагмента кода:

import os
def list_files_with_extension(directory, extension):
    files = os.listdir(directory)
    filtered_files = [file for file in files if file.endswith(extension)]
    return filtered_files
# Usage example
directory_path = '/path/to/directory'
file_extension = '.txt'
filtered_files = list_files_with_extension(directory_path, file_extension)
print(filtered_files)

Метод 2: использование модуля glob
Модуль glob Python — еще один мощный инструмент для работы с шаблонами файлов. Он позволяет находить файлы по подстановочным знакам и сопоставлению с образцом. Вот как его можно использовать для вывода списка файлов с определенным расширением:

import glob
def list_files_with_extension(directory, extension):
    pattern = directory + '/*' + extension
    filtered_files = glob.glob(pattern)
    return filtered_files
# Usage example
directory_path = '/path/to/directory'
file_extension = '.txt'
filtered_files = list_files_with_extension(directory_path, file_extension)
print(filtered_files)

Метод 3: использование модуля pathlib
Модуль pathlib, представленный в Python 3.4, обеспечивает объектно-ориентированный подход для обработки путей и файлов. Он предлагает удобный способ составить список файлов с определенным расширением. Вот пример:

from pathlib import Path
def list_files_with_extension(directory, extension):
    path = Path(directory)
    filtered_files = [file.name for file in path.iterdir() if file.name.endswith(extension)]
    return filtered_files
# Usage example
directory_path = '/path/to/directory'
file_extension = '.txt'
filtered_files = list_files_with_extension(directory_path, file_extension)
print(filtered_files)

В этой статье мы рассмотрели три различных метода вывода списка файлов с определенным расширением в Python. Мы использовали возможности os.listdir(), гибкость модуля glob и объектно-ориентированную природу модуля pathlib. Каждый метод имеет свои преимущества, поэтому выберите тот, который лучше всего соответствует вашим потребностям. Благодаря этим методам в вашем арсенале программирования вы сможете эффективно обрабатывать файлы и каталоги в своих проектах Python.

Не забудьте адаптировать примеры кода к вашему конкретному варианту использования, указав правильный путь к каталогу и расширение файла. Приятного кодирования!