Pythonで指定された深さのディレクトリを一覧表示する

Question

指定されたパスと深さでディレクトリを持つリストを返す関数が必要です。私はos.walkをかなり使っていますが、深さなどを数えるときにコードが醜いようになってきました。

実際に最も「きちんとした」実装は何ですか？

Answer 1

深さが固定されている場合は、globは良いアイデアです：

import glob,os.path 
filesDepth3 = glob.glob('*/*/*') 
dirsDepth3 = filter(lambda f: os.path.isdir(f), filesDepth3) 

そうでなければ、それはos.walkを使うにはあまりにも難しいことではありません。

import os,string 
path = '.' 
path = os.path.normpath(path) 
res = [] 
for root,dirs,files in os.walk(path, topdown=True): 
    depth = root[len(path) + len(os.path.sep):].count(os.path.sep) 
    if depth == 2: 
     # We're currently two directories in, so all subdirs have depth 3 
     res += [os.path.join(root, d) for d in dirs] 
     dirs[:] = [] # Don't recurse any deeper 
print(res) 

Answer 2

これはまさにきちんとではありませんUNIXのようなOSの場合は、 "find"のようなシステムツールを利用し、それを外部プログラムとして実行することもできます。

from subprocess import call 
call(["find", "-maxdepth", "2", "-type", "d"]) 

出力を何らかの文字列変数にリダイレクトして、後で処理することができます。

Answer 3

私は本当にphihagの答えが好きです。私は自分のニーズに合わせてそれを適応させました。

def fileNamesRetrieve(top, maxDepth, fnMask ): 
    for d in range(1, maxDepth+1): 
     maxGlob = "/".join("*" * d) 
     topGlob = os.path.join(top, maxGlob) 
     allFiles = glob.glob(topGlob) 
     if fnmatch.fnmatch(os.path.basename(f), fnMask): 
      yield f 

批判歓迎：

import fnmatch,glob 
def fileNamesRetrieve(top, maxDepth, fnMask ): 
    someFiles = [] 
    for d in range(1, maxDepth+1): 
     maxGlob = "/".join("*" * d) 
     topGlob = os.path.join(top, maxGlob) 
     allFiles = glob.glob(topGlob) 
     someFiles.extend([ f for f in allFiles if fnmatch.fnmatch(os.path.basename(f), fnMask) ]) 
    return someFiles 

私はまた、このようなもので発電させることができると思います。