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

들어 파이썬 <2.5 = 사용 :

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

작동 원리

os.walk은 생성기이며 호출 next은 3 개의 튜플 (dirpath, dirnames, filenames) 형식으로 첫 번째 결과를 얻습니다. 따라서 [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)
    ]


답변

이것은 적어도 리눅스에서 작동하는 것 같습니다.

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 ()의 필터링은 매우 빠릅니다.