Список файлов ТОЛЬКО в текущем каталоге Python: без подкаталогов

Быстрый ответ

Для формирования списка имен только файлов, находящихся в текущей директории, можно использовать следующий код:

import os
print([f for f in os.listdir('.') if os.path.isfile(f)])

Такой подход обеспечивает получение понятного списка файлов благодаря применению генератора списков и функции os.path.isfile().

Методы определения содержимого каталога: модули 'os' и 'pathlib'

Несмотря на то, что функция os.listdir весьма удобна, для более эффективной работы рекомендуется применять функцию os.scandir(), особенно с Python версией 3.5 и выше. Функция os.scandir() предоставляет итератор для сканирования, который увеличивает производительность за счёт сокращения числа системных вызовов.

Рассмотрим теперь модуль pathlib, представляющий объектно-ориентированный подход к обработке файловых путей:

from pathlib import Path

# Получаем список файлов с помощью pathlib и генератора
files_in_current_directory = (entry for entry in Path('.').iterdir() if entry.is_file())
print(list(files_in_current_directory))

Модуль pathlib также предлагает метод is_file(), аналог os.path.isfile(), для фильтрации только файлов:

# Фильтруем файлы, как при традиционном отборе
files_in_current_directory = filter(lambda p: p.is_file(), Path('.').iterdir())
print(list(files_in_current_directory))

Приведение в порядок: Сортировка файлов по расширениям

Классифицируем файлы по их расширениям с помощью практичного генератора словарей:

# Как библиотекарь, разделяем файлы по категориям
files_by_extension = {
    ext: [file for file in os.listdir('.') if file.endswith(ext)]
    for ext in ['.jpg', '.txt', '.py', '.md']  # Добавляйте расширения при необходимости
}

Для перемещения файлов в соответствии с их расширениями используем модуль shutil:

import shutil

# По образу гроссмейстера в шахматах, перемещаем файлы по категориям
for ext, files in files_by_extension.items():
    for file in files:
        shutil.move(file, f"destination_folder/{file}")

Всегда применяйте обработку исключений для предотвращения возникновения ошибок, которые могут произойти из-за отсутствия файлов или нюансов файловой системы:

try:
    # Обработка операций с файлами с осторожностью
except OSError as e:
    print(f"Ошибка: {e.strerror}")

Простота превыше всего

Если вы цените простоту и хотите избежать излишних импортов, можно использовать модуль glob:

from glob import glob

print(glob('*.py'))  # Выводит все Python-файлы в текущем каталоге

Визуализация

Представьте директории в виде птиц (🐦) и файлы как бумаги (📄). Ваша задача – выбрать бумаги:

Изображение:  🐦📄📄🐦🐦📄📄🐦
Отфильтровано: [📄, 📄, 📄, 📄]

И вуаля, мы успешно обнаружили все файлы — это прекрасно! 🎉

Управление рекурсией: Методичный обход

Функция os.walk позволяет рекурсивно обходить каталоги и обладает высокой мощностью. Как известно, с большой мощностью идёт большая ответственность:

for root, dirs, files in os.walk('.', topdown=True):
    # 'dirs.clear()' работает как барьер, останавливающий рекурсию директорий
    dirs.clear()
    print(files)
    break  # Завершаем цикл, хотя скрипт пока что все еще работает!

Данный код ограничивает рекурсию, очищая список dirs, и прекращает обход при первой итерации.