จะแสดงเฉพาะไดเรกทอรีระดับบนสุดใน 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]

สำหรับPython <= 2.5 ให้ใช้:

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

วิธีการทำงาน

os.walkเป็นเครื่องกำเนิดไฟฟ้าและการโทรnextจะได้ผลลัพธ์แรกในรูปแบบของ 3-tuple (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)
    ]

ดูเหมือนว่าจะได้ผลเช่นกัน (อย่างน้อยใน linux):

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

เพียงเพื่อเพิ่มว่าการใช้ os.listdir () ไม่ได้"ใช้เวลามากของการประมวลผลเทียบ os.walk ง่ายมาก (). ถัดไป () [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 () เร็วขึ้นเล็กน้อย

วิธีที่ง่ายและสง่างามมากคือการใช้สิ่งนี้:

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

เรียกใช้สคริปต์นี้ในโฟลเดอร์เดียวกับที่คุณต้องการชื่อโฟลเดอร์มันจะให้ชื่อโฟลเดอร์ที่ถูกต้องเท่านั้น (ซึ่งก็ไม่มีเส้นทางแบบเต็มของโฟลเดอร์)

การใช้ความเข้าใจในรายการ

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

ฉันคิดว่ามันเป็นวิธีที่ง่ายที่สุด

เป็นมือใหม่ที่นี่ฉันยังไม่สามารถแสดงความคิดเห็นได้โดยตรง แต่นี่คือการแก้ไขเล็กน้อยที่ฉันต้องการเพิ่มในส่วนต่อไปนี้ของคำตอบของΤΖΩΤΖΙΟΥ :

หากคุณต้องการชื่อพา ธ แบบเต็มให้ใช้ฟังก์ชันนี้:

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

สำหรับผู้ที่ยังอยู่ใน python <2.4 : โครงสร้างด้านในจะต้องเป็นรายการแทนที่จะเป็นทูเปิลดังนั้นจึงควรอ่านดังนี้:

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

มิฉะนั้นจะมีข้อผิดพลาดทางไวยากรณ์

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

สำหรับรายการชื่อพา ธ แบบเต็มฉันชอบเวอร์ชันนี้มากกว่าโซลูชันอื่น ๆที่นี่:

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

ตัวเลือกที่ปลอดภัยกว่าที่ไม่ล้มเหลวเมื่อไม่มีไดเร็กทอรี

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

ชอบมั้ย?

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

งูหลาม 3.4 แนะนำโมดูลเข้าไปในห้องสมุดมาตรฐานที่ให้บริการเชิงวัตถุวิธีการเส้นทางการจัดการระบบแฟ้ม:pathlib

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