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タプル（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)。関数呼び出しが1つ少なくなり、移植性も高くなります。

したがって、答えを完成させるには、フォルダ内のディレクトリのリストを取得します。

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（）。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（）のフィルタリングはわずかに高速です。

非常にシンプルでエレガントな方法は、これを使用することです。

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

このスクリプトを、フォルダー名が必要な同じフォルダーで実行します。これにより、直接のフォルダー名のみが正確に示されます（フォルダーの完全パスがない場合）。

リスト内包表記を使用して、

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

一番簡単な方法だと思います

ここでは初心者なので、まだ直接コメントすることはできませんが、hereΩΤΖΙΟΥの回答の次の部分に追加したい小さな修正があります。

完全パス名を使用する場合は、次の関数を使用してください：

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

Python 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