Есть ли способ получить подкаталоги в Python без необходимости перебора по всем файлам?

Question

Я ищу способ перечислить подкаталоги, содержащиеся в текущем рабочем каталоге, однако мне не удалось найти способ, который не перебирает все файлы.

По существу, если у меня есть папка с большим количеством файлов и 2 папки, мне нужен метод, который может быстро вернуть список, содержащий имена 2-х папок, без необходимости также сканировать все файлы.

Есть ли способ сделать это в Python?

Редактировать: Я должен уточнить, что мой вопрос касается производительности извлечения каталогов. Я уже знаю несколько способов получить каталоги, но они все замедлились, если в рабочем каталоге есть куча файлов.

Answer 1

Невозможно только восстановить каталоги из операционной системы. Вы должны отфильтровать результаты. Хотя, похоже, использование os.scandir повышает производительность порядка magnitude (см. Тесты) по сравнению с os.listdir и старшей версией os.walk, поскольку она позволяет избежать получения чего-либо, кроме метаданных, где это возможно. Если вы используете 3.5, он уже интегрирован в стандартную библиотеку. В противном случае, похоже, вам нужно использовать пакет scandir.

Чтобы отфильтровать результаты от os.scandir

ds = [e.name() for e in os.scandir('.') if e.is_dir()] 

Согласно документации, walk осуществляется в терминах scandir, которая также дает тот же ускорение.

Answer 2

Не уверен, есть ли какие-либо прямые стандартные функции, которые сделают это для вас. Но Вы можете использовать os.walk() для этого, каждая итерация os.walk() возвращает кортеж формата -

(dirpath, dirnames, filenames) 

Где dirpath это каталог будучи шел в настоящее время, dirnames содержит каталоги внутри dirpath и filenames содержит файлы внутри него ,

Вы можете просто позвонить по телефону next(os.walk()), чтобы получить приведенный выше кортеж для каталога, а второй элемент (индекс - 1) в этом кортеже будет подпапками внутри каталога.

код -

direcs = next(os.walk('.'))[1] 

direcs в конце будет список подпапок текущей папки. Вы также можете указать другую папку, чтобы получить список папок внутри нее.