Como listar apenas diretórios de nível superior em Python?

Quero poder listar apenas os diretórios dentro de alguma pasta. Isso significa que eu não quero nomes de arquivos listados, nem subpastas adicionais.

Vamos ver se um exemplo ajuda. No diretório atual, temos:

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

No entanto, não quero nomes de arquivos listados. Também não quero subpastas como \ Lib \ curses. Essencialmente, o que eu quero funciona com o seguinte:

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

No entanto, estou me perguntando se existe uma maneira mais simples de obter os mesmos resultados. Tenho a impressão de que usar o os.walk apenas para retornar o nível superior é ineficiente / demais.

Filtre o resultado usando os.path.isdir () (e use os.path.join () para obter o caminho 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']

os.walk

Use os.walkcom a nextfunção item:

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

Para Python <= 2.5, use:

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

Como isso funciona

os.walké um gerador e a chamada nextobterá o primeiro resultado na forma de três tuplas (caminho de rota, nome de diretório, nome de arquivo). Assim, o [1]índice retorna apenas o dirnamesdessa tupla.

Filtre a lista usando os.path.isdir para detectar diretórios.

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

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

Observe que, em vez de fazer os.listdir(os.getcwd()), é preferível fazer os.listdir(os.path.curdir). Menos uma chamada de função e é tão portátil.

Portanto, para concluir a resposta, obtenha uma lista de diretórios em uma pasta:

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

Se você preferir nomes de caminho completos, use esta função:

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

Isso parece funcionar também (pelo menos no linux):

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

Apenas para acrescentar que o uso de os.listdir () não "exige muito processamento versus muito simples os.walk (). Next () [1]" . Isso ocorre porque o os.walk () usa os.listdir () internamente. De fato, se você testá-los 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

A filtragem de os.listdir () é muito ligeiramente mais rápida.

Uma maneira muito mais simples e elegante é usar isso:

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

Execute esse script na mesma pasta para a qual você deseja nomes de pastas. Ele fornecerá exatamente o nome imediato das pastas (isso também sem o caminho completo das pastas).

Usando a compreensão da lista,

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

Eu acho que é a maneira mais simples

sendo um novato aqui, ainda não posso comentar diretamente, mas aqui está uma pequena correção que gostaria de adicionar à parte seguinte do resposta ΤΖΩΤΖΙΟΥ :

Se você preferir nomes de caminho completos, use esta função:

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 quem ainda está em python <2.4 : a construção interna precisa ser uma lista em vez de uma tupla e, portanto, deve ser assim:

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

caso contrário, obtém-se um erro de sintaxe.

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

Para uma lista de nomes de caminhos completos, prefiro esta versão às outras soluções aqui:

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

Uma opção mais segura que não falha quando não há diretório.

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

Igual a?

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

O Python 3.4 introduziu o pathlibmódulo na biblioteca padrão, que fornece uma abordagem orientada a objetos para lidar com os caminhos do sistema de arquivos:

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