Як перелічити лише каталоги верхнього рівня в Python?

Я хочу мати змогу перераховувати лише каталоги всередині якоїсь папки. Це означає, що я не хочу перелічувати імена файлів, а також не хочу додаткових підпапок.

Подивимось, чи допомагає приклад. У поточному каталозі ми маємо:

>>> os.listdir(os.getcwd())
['cx_Oracle-doc', 'DLLs', 'Doc', 'include', 'Lib', 'libs', 'LICENSE.txt', 'mod_p
ython-wininst.log', 'NEWS.txt', 'pymssql-wininst.log', 'python.exe', 'pythonw.ex
e', 'README.txt', 'Removemod_python.exe', 'Removepymssql.exe', 'Scripts', 'tcl',
 'Tools', 'w9xpopen.exe']

Однак я не хочу перелічувати імена файлів. Також я не хочу підпапок, таких як \ Lib \ curses. По суті, те, що я хочу, працює з наступним:

>>> for root, dirnames, filenames in os.walk('.'):
...     print dirnames
...     break
...
['cx_Oracle-doc', 'DLLs', 'Doc', 'include', 'Lib', 'libs', 'Scripts', 'tcl', 'Tools']

Однак мені цікаво, чи існує простіший спосіб досягти однакових результатів. У мене складається враження, що використовувати os.walk лише для повернення верхнього рівня неефективно / занадто багато.

Фільтруйте результат за допомогою os.path.isdir () (та використовуйте os.path.join (), щоб отримати реальний шлях):

>>> [ name for name in os.listdir(thedir) if os.path.isdir(os.path.join(thedir, name)) ]
['ctypes', 'distutils', 'encodings', 'lib-tk', 'config', 'idlelib', 'xml', 'bsddb', 'hotshot', 'logging', 'doc', 'test', 'compiler', 'curses', 'site-packages', 'email', 'sqlite3', 'lib-dynload', 'wsgiref', 'plat-linux2', 'plat-mac']

os.walk

Використовувати os.walkз nextфункцією елемента:

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

Для Python <= 2,5 використовуйте:

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

Як це працює

os.walkє генератором і виклик nextотримає перший результат у вигляді 3-кортежу (dirpath, dirnames, filename). Таким чином, [1]індекс повертає лише dirnamesз цього кортежу.

Фільтруйте список, використовуючи os.path.isdir для виявлення каталогів.

filter(os.path.isdir, os.listdir(os.getcwd()))

directories=[d for d in os.listdir(os.getcwd()) if os.path.isdir(d)]

Зауважте, що замість цього робити os.listdir(os.getcwd())краще os.listdir(os.path.curdir). Один менш функціональний дзвінок, і він настільки портативний.

Отже, щоб заповнити відповідь, щоб отримати список каталогів у папці:

def listdirs(folder):
    return [d for d in os.listdir(folder) if os.path.isdir(os.path.join(folder, d))]

Якщо ви віддаєте перевагу повних імен, використовуйте цю функцію:

def listdirs(folder):
    return [
        d for d in (os.path.join(folder, d1) for d1 in os.listdir(folder))
        if os.path.isdir(d)
    ]

Це, здається, теж працює (принаймні, на Linux):

import glob, os
glob.glob('*' + os.path.sep)

Просто додамо, що використання os.listdir () не «займає багато обробки та дуже простого os.walk (). Next () [1]» . Це тому, що os.walk () використовує os.listdir () внутрішньо. Насправді, якщо ви протестуєте їх разом:

>>>> import timeit
>>>> timeit.timeit("os.walk('.').next()[1]", "import os", number=10000)
1.1215229034423828
>>>> timeit.timeit("[ name for name in os.listdir('.') if os.path.isdir(os.path.join('.', name)) ]", "import os", number=10000)
1.0592019557952881

Фільтрація os.listdir () відбувається дуже дещо швидше.

Дуже простішим і елегантним способом є використання цього:

 import os
 dir_list = os.walk('.').next()[1]
 print dir_list

Запустіть цей скрипт у тій самій папці, для якої ви хочете імена папок. Він дасть вам саме те саме ім'я папок (що теж без повного шляху папок).

Використовуючи розуміння списку,

[a for a in os.listdir() if os.path.isdir(a)]

Я думаю, що це найпростіший спосіб

будучи новачком тут, я ще не можу прямо коментувати, але ось невелике виправлення, яке я хотів би додати до наступної частини відповіді ΤΖΩΤΖΙΟΥ :

Якщо ви віддаєте перевагу повних імен, використовуйте цю функцію:

def listdirs(folder):  
  return [
    d for d in (os.path.join(folder, d1) for d1 in os.listdir(folder))
    if os.path.isdir(d)
]

для тих, хто все ще знаходиться на python <2.4 : внутрішня конструкція повинна бути списком замість кортежу, і тому слід читати так:

def listdirs(folder):  
  return [
    d for d in [os.path.join(folder, d1) for d1 in os.listdir(folder)]
    if os.path.isdir(d)
  ]

інакше виникає синтаксична помилка.

[x for x in os.listdir(somedir) if os.path.isdir(os.path.join(somedir, x))]

Для переліку повних імен шляхів я віддаю перевагу цій версії перед іншими рішеннями тут:

def listdirs(dir):
    return [os.path.join(os.path.join(dir, x)) for x in os.listdir(dir) 
        if os.path.isdir(os.path.join(dir, x))]

scanDir = "abc"
directories = [d for d in os.listdir(scanDir) if os.path.isdir(os.path.join(os.path.abspath(scanDir), d))]

Більш безпечний варіант, який не виходить з ладу, коли немає каталогу.

def listdirs(folder):
    if os.path.exists(folder):
         return [d for d in os.listdir(folder) if os.path.isdir(os.path.join(folder, d))]
    else:
         return []

Як так?

>>>> [path for path in os.listdir(os.getcwd()) if os.path.isdir(path)]

Python 3.4 вводиться в pathlibмодулі в стандартну бібліотеку, яка забезпечує об'єктно - орієнтований підхід для обробки файлової системи шляхів:

from pathlib import Path

p = Path('./')
[f for f in p.iterdir() if f.is_dir()]

-- This will exclude files and traverse through 1 level of sub folders in the root

def list_files(dir):
    List = []
    filterstr = ' '
    for root, dirs, files in os.walk(dir, topdown = True):
        #r.append(root)
        if (root == dir):
            pass
        elif filterstr in root:
            #filterstr = ' '
            pass
        else:
            filterstr = root
            #print(root)
            for name in files:
                print(root)
                print(dirs)
                List.append(os.path.join(root,name))
            #print(os.path.join(root,name),"\n")
                print(List,"\n")

    return List