Wie liste ich nur Verzeichnisse der obersten Ebene in Python auf?

Ich möchte nur die Verzeichnisse in einem Ordner auflisten können. Dies bedeutet, dass ich weder Dateinamen noch zusätzliche Unterordner auflisten möchte.

Mal sehen, ob ein Beispiel hilft. Im aktuellen Verzeichnis haben wir:

>>> 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']

Ich möchte jedoch nicht, dass Dateinamen aufgelistet werden. Ich möchte auch keine Unterordner wie \ Lib \ curses. Im Wesentlichen funktioniert das, was ich möchte, mit Folgendem:

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

Ich frage mich jedoch, ob es einen einfacheren Weg gibt, die gleichen Ergebnisse zu erzielen. Ich habe den Eindruck, dass die Verwendung von os.walk nur zur Rückgabe der obersten Ebene ineffizient / zu viel ist.

Filtern Sie das Ergebnis mit os.path.isdir () (und verwenden Sie os.path.join (), um den tatsächlichen Pfad zu erhalten):

>>> [ 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

Verwendung os.walkmit nextItem-Funktion:

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

Für Python <= 2.5 verwenden Sie:

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

Wie das funktioniert

os.walkist ein Generator und der Aufruf nexterhält das erste Ergebnis in Form eines 3-Tupels (dirpath, dirnames, filenames). Somit gibt der [1]Index nur das dirnamesvon diesem Tupel zurück.

Filtern Sie die Liste mit os.path.isdir, um Verzeichnisse zu erkennen.

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

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

Beachten Sie, dass dies os.listdir(os.getcwd())vorzuziehen ist os.listdir(os.path.curdir). Ein Funktionsaufruf weniger, und es ist genauso portabel.

Um die Antwort zu vervollständigen und eine Liste der Verzeichnisse in einem Ordner zu erhalten:

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

Wenn Sie vollständige Pfadnamen bevorzugen, verwenden Sie diese Funktion:

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

Dies scheint auch zu funktionieren (zumindest unter Linux):

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

Nur um hinzuzufügen, dass die Verwendung von os.listdir () nicht "viel Verarbeitung im Vergleich zu sehr einfachem os.walk (). Next () [1] erfordert" . Dies liegt daran, dass os.walk () os.listdir () intern verwendet. In der Tat, wenn Sie sie zusammen testen:

>>>> 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

Das Filtern von os.listdir () ist etwas schneller.

Eine sehr viel einfachere und elegantere Möglichkeit besteht darin, Folgendes zu verwenden:

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

Führen Sie dieses Skript in demselben Ordner aus, für den Sie Ordnernamen möchten. Es gibt Ihnen nur genau den Namen des unmittelbaren Ordners (auch dies ohne den vollständigen Pfad der Ordner).

Listenverständnis verwenden,

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

Ich denke, es ist der einfachste Weg

Da ich hier ein Neuling bin, kann ich noch keinen direkten Kommentar abgeben, aber hier ist eine kleine Korrektur, die ich dem folgenden Teil der Antwort von ΤΖΩΤΖΙΟΥ hinzufügen möchte :

Wenn Sie vollständige Pfadnamen bevorzugen, verwenden Sie diese Funktion:

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

für diejenigen, die noch auf Python <2.4 sind : Das innere Konstrukt muss eine Liste anstelle eines Tupels sein und sollte daher folgendermaßen lauten:

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

Andernfalls wird ein Syntaxfehler angezeigt.

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

Für eine Liste der vollständigen Pfadnamen ziehe ich diese Version den anderen Lösungen hier vor:

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))]

Eine sicherere Option, die nicht fehlschlägt, wenn kein Verzeichnis vorhanden ist.

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 []

Wie so?

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

Python 3.4 führte das pathlibModul in die Standardbibliothek ein, die einen objektorientierten Ansatz für die Verarbeitung von Dateisystempfaden bietet:

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