¿Cómo enumerar solo los directorios de nivel superior en Python?

Question

Quiero ser capaz de enumerar solo los directorios dentro de alguna carpeta. Esto significa que no quiero que aparezcan los nombres de los archivos, ni quiero otras subcarpetas.

Veamos si un ejemplo ayuda. En el directorio actual tenemos:

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

Sin embargo, no quiero los nombres de los archivos listados. Tampoco quiero subcarpetas como \ Lib \ maldiciones. Esencialmente lo que quiero funciona con las siguientes:

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

Sin embargo, me pregunto si hay una manera más sencilla de lograr los mismos resultados. Me da la impresión de que usar os.walk solo para devolver el nivel superior es ineficiente/demasiado.

Answer 1

Filtrar el resultado utilizando os.path.isdir() (y utilizar os.path.join() para obtener la ruta real):

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

Answer 2

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

Answer 3

Me gusta?

>>> [ruta para la ruta en os.listdir (os.getcwd()) si os.path.isdir (ruta de acceso)]

Answer 4

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

Answer 5

Filtrar la lista mediante os.path.isdir para detectar directorios.

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

Answer 6

Tenga en cuenta que, en vez de hacer os.listdir(os.getcwd()), es preferible hacer os.listdir(os.path.curdir). Llamada a una función menos, y es tan portátil.

lo tanto, para completar la respuesta, para obtener una lista de directorios en una carpeta:

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

Si prefiere rutas completas, a continuación, utilizar esta función:

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

Answer 7

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

o con Python 3 use

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

como @Andre Soares dijo en el comentario a continuación.

Answer 8

ser un novato aquí que no puedo comentar todavía directamente pero aquí es una pequeña corrección Me gustaría añadir a la siguiente parte de ΤΖΩΤΖΙΟΥ's answer:

Si prefiere rutas completas, a continuación, utilizar esta función :

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

para los que siguen en pitón < 2,4: el constructo interior necesita ser una lista en lugar de una tupla y por lo tanto debe leer como ésimo es:

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

De lo contrario, uno obtiene un error de sintaxis.

Answer 9

Solo para agregar que usar os.listdir() no hace "tomar un montón de procesamiento vs muy simple os.walk(). Next() [1]". Esto se debe a que os.walk() usa os.listdir() internamente. De hecho, si los prueba juntos:

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

El filtrado de os.listdir() es muy ligeramente más rápido.

Answer 10

Para obtener una lista de nombres de ruta completos Yo prefiero esta versión a la otra solutions aquí:

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

Answer 11

Una manera mucho más simple y elegante es utilizar esto:

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

Ejecutar este proceso en la misma carpeta para la que desea nombres de carpeta. Le dará exactamente el nombre de las carpetas inmediatas solamente (también sin la ruta completa de las carpetas).

Answer 12

Esto parece funcionar también (al menos en Linux):

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

Answer 13

Una opción más segura que no falla cuando no hay ningún directorio.

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

Answer 14

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

Answer 15

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