Làm thế nào để liệt kê chỉ các thư mục cấp cao nhất trong Python?

Tôi muốn chỉ có thể liệt kê các thư mục trong một số thư mục. Điều này có nghĩa là tôi không muốn tên tệp được liệt kê, tôi cũng không muốn các thư mục con bổ sung.

Hãy xem nếu một ví dụ giúp. Trong thư mục hiện tại, chúng tôi có:

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

Tuy nhiên, tôi không muốn tên tập tin được liệt kê. Tôi cũng không muốn các thư mục con như \ Lib \ curses. Về cơ bản những gì tôi muốn làm việc với những điều sau đây:

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

Tuy nhiên, tôi tự hỏi liệu có cách nào đơn giản hơn để đạt được kết quả tương tự. Tôi có ấn tượng rằng việc sử dụng os.walk chỉ để trả về mức cao nhất là không hiệu quả / quá nhiều.

Lọc kết quả bằng os.path.itorir () (và sử dụng os.path.join () để có được đường dẫn thực):

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

Sử dụng os.walkvới nextchức năng mục:

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

Đối với Python <= 2.5 sử dụng:

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

Làm thế nào điều này hoạt động

os.walklà một trình tạo và gọi nextsẽ nhận được kết quả đầu tiên dưới dạng 3-tuple (dirpath, dirnames, filenames). Do đó, [1]chỉ mục trả về chỉ dirnamestừ bộ dữ liệu đó.

Lọc danh sách bằng os.path.itorir để phát hiện các thư mục.

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

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

Lưu ý rằng, thay vì làm os.listdir(os.getcwd()), tốt hơn là nên làm os.listdir(os.path.curdir). Một cuộc gọi ít chức năng hơn và đó là cuộc gọi di động.

Vì vậy, để hoàn thành câu trả lời, để có được danh sách các thư mục trong một thư mục:

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

Nếu bạn thích tên đường dẫn đầy đủ, sau đó sử dụng chức năng này:

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

Điều này dường như cũng hoạt động (ít nhất là trên linux):

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

Chỉ cần thêm rằng sử dụng os.listdir () không "mất nhiều công sức xử lý so với os.walk (). Next () [1]" rất đơn giản . Điều này là do os.walk () sử dụng os.listdir () trong nội bộ. Trong thực tế nếu bạn kiểm tra chúng cùng nhau:

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

Quá trình lọc os.listdir () nhanh hơn một chút.

Một cách rất đơn giản và thanh lịch là sử dụng điều này:

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

Chạy tập lệnh này trong cùng một thư mục mà bạn muốn có tên thư mục. Nó sẽ chỉ cung cấp cho bạn chính xác tên thư mục ngay lập tức (cũng không có đường dẫn đầy đủ của các thư mục).

Sử dụng hiểu danh sách,

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

Tôi nghĩ đó là cách đơn giản nhất

là một người mới ở đây tôi chưa thể bình luận trực tiếp nhưng đây là một chỉnh sửa nhỏ tôi muốn thêm vào phần sau của câu trả lời :

Nếu bạn thích tên đường dẫn đầy đủ, sau đó sử dụng chức năng này:

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

đối với những người vẫn còn trên python <2.4 : cấu trúc bên trong cần phải là một danh sách thay vì một tuple và do đó nên đọc như sau:

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

nếu không thì sẽ bị lỗi cú pháp.

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

Để biết danh sách tên đường dẫn đầy đủ, tôi thích phiên bản này hơn các giải pháp khác tại đây:

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

Một tùy chọn an toàn hơn mà không thất bại khi không có thư mục.

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

Thích như vậy?

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

Python 3.4 giới thiệu các pathlibmô-đun vào thư viện chuẩn, cung cấp một cách tiếp cận hướng đối tượng để xử lý đường dẫn hệ thống tập tin:

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