Bagaimana cara mendaftar hanya direktori tingkat atas di Python?

Saya ingin hanya dapat mendaftar direktori di dalam beberapa folder. Ini berarti saya tidak ingin nama file terdaftar, saya juga tidak ingin sub-folder tambahan.

Mari kita lihat apakah sebuah contoh membantu. Dalam direktori saat ini kami memiliki:

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

Namun, saya tidak ingin nama file terdaftar. Saya juga tidak ingin sub-folder seperti \ Lib \ curses. Pada dasarnya apa yang saya inginkan berfungsi dengan yang berikut:

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

Namun, saya bertanya-tanya apakah ada cara yang lebih sederhana untuk mencapai hasil yang sama. Saya mendapat kesan bahwa menggunakan os.walk hanya untuk mengembalikan level atas tidak efisien / terlalu banyak.

Memfilter hasil menggunakan os.path.isdir () (dan menggunakan os.path.join () untuk mendapatkan jalur nyata):

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

Gunakan os.walkdengan nextfungsi item:

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

Untuk Python <= 2,5 gunakan:

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

Bagaimana ini bekerja?

os.walkadalah generator dan panggilan nextakan mendapatkan hasil pertama dalam bentuk 3-tuple (dirpath, dirnames, nama file). Dengan demikian [1]indeks hanya mengembalikan dirnamesdari tuple itu.

Saring daftar menggunakan os.path.isdir untuk mendeteksi direktori.

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

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

Perhatikan bahwa, alih-alih melakukan os.listdir(os.getcwd()), itu lebih baik dilakukanos.listdir(os.path.curdir) . Satu panggilan fungsi kurang, dan itu sebagai portabel.

Jadi, untuk melengkapi jawabannya, untuk mendapatkan daftar direktori di folder:

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

Jika Anda lebih suka nama path lengkap, gunakan fungsi ini:

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

Ini sepertinya berfungsi juga (setidaknya di linux):

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

Hanya untuk menambahkan bahwa menggunakan os.listdir () tidak "mengambil banyak pemrosesan vs os.walk sangat sederhana (). Next () [1]" . Ini karena os.walk () menggunakan os.listdir () secara internal. Bahkan jika Anda mengujinya bersama:

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

Pemfilteran os.listdir () sangat sedikit lebih cepat.

Cara yang sangat sederhana dan elegan adalah dengan menggunakan ini:

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

Jalankan skrip ini di folder yang sama yang Anda inginkan nama folder. Ini akan memberi Anda persis nama folder langsung saja (itu juga tanpa path lengkap folder).

Menggunakan pemahaman daftar,

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

Saya pikir ini adalah cara paling sederhana

menjadi pemula di sini saya belum bisa berkomentar secara langsung tetapi di sini ada koreksi kecil saya ingin menambahkan ke bagian berikut dari jawaban ΤΖΩΤΖΙΟΥ :

Jika Anda lebih suka nama path lengkap, gunakan fungsi ini:

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

bagi yang masih menggunakan python <2.4 : konstruk dalam perlu daftar bukan tupel dan karena itu harus membaca seperti ini:

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

jika tidak, orang akan mendapatkan kesalahan sintaksis.

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

Untuk daftar nama jalur lengkap, saya lebih suka versi ini daripada solusi lain di sini:

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

Opsi yang lebih aman yang tidak gagal saat tidak ada direktori.

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

Seperti begitu?

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

Python 3.4 diperkenalkan pada pathlibmodul ke dalam perpustakaan standar, yang menyediakan pendekatan berorientasi objek untuk jalur menangani filesystem:

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