以下のディレクトリを掘り下げることなくos.walk

指定os.walkしたディレクトリ内のファイルのみを返すように制限するにはどうすればよいですか？

def _dir_list(self, dir_name, whitelist):
    outputList = []
    for root, dirs, files in os.walk(dir_name):
        for f in files:
            if os.path.splitext(f)[1] in whitelist:
                outputList.append(os.path.join(root, f))
            else:
                self._email_to_("ignore")
    return outputList

walklevel関数を使用します。

import os

def walklevel(some_dir, level=1):
    some_dir = some_dir.rstrip(os.path.sep)
    assert os.path.isdir(some_dir)
    num_sep = some_dir.count(os.path.sep)
    for root, dirs, files in os.walk(some_dir):
        yield root, dirs, files
        num_sep_this = root.count(os.path.sep)
        if num_sep + level <= num_sep_this:
            del dirs[:]

これはと同じように機能しos.walkますがlevel、再帰の深さを示すパラメーターを渡すことができます。

os.walkは使用しないでください。

例：

import os

root = "C:\\"
for item in os.listdir(root):
    if os.path.isfile(os.path.join(root, item)):
        print item

ソリューションは実際には非常に単純だと思います。

使用する

break

forループの最初の反復のみを実行するには、よりエレガントな方法が必要です。

for root, dirs, files in os.walk(dir_name):
    for f in files:
        ...
        ...
    break
...

初めてos.walkを呼び出すと、現在のディレクトリのチューリップが返され、次に次のループで次のディレクトリのコンテンツが返されます。

オリジナルのスクリプトを取り、改行を追加するだけです。

def _dir_list(self, dir_name, whitelist):
    outputList = []
    for root, dirs, files in os.walk(dir_name):
        for f in files:
            if os.path.splitext(f)[1] in whitelist:
                outputList.append(os.path.join(root, f))
            else:
                self._email_to_("ignore")
        break
    return outputList

使用する提案listdirは良いものです。Python 2での質問に対する直接の答えはroot, dirs, files = os.walk(dir_name).next()です。

同等のPython 3構文は次のとおりです。 root, dirs, files = next(os.walk(dir_name))

os.listdir()whichを使用すると、特定のディレクトリ内の（ファイルとディレクトリの両方の）名前のリストを返します。ファイルとディレクトリを区別する必要がある場合はos.stat()、それぞれの名前を呼び出します。

トップディレクトリ以外の複雑な要件がある場合（たとえば、VCSディレクトリを無視するなど）、ディレクトリのリストを変更して、os.walkが再帰しないようにすることもできます。

つまり：

def _dir_list(self, dir_name, whitelist):
    outputList = []
    for root, dirs, files in os.walk(dir_name):
        dirs[:] = [d for d in dirs if is_good(d)]
        for f in files:
            do_stuff()

注-単に再バインドするのではなく、リストを変更するように注意してください。明らかに、os.walkは外部の再バインドを認識していません。

for path, dirs, files in os.walk('.'):
    print path, dirs, files
    del dirs[:] # go only one level deep

と同じ考え方ですがlistdir、短くなっています。

[f for f in os.listdir(root_dir) if os.path.isfile(os.path.join(root_dir, f))]

私の2ペンスを投げ込むように感じました。

baselevel = len(rootdir.split("\\"))
for subdirs, dirs, files in os.walk(rootdir):
    curlevel = len(subdirs.split("\\"))
    if curlevel <= baselevel + 1:
        [do stuff]

Python 3では、これを行うことができました。

import os
dir = "/path/to/files/"

#List all files immediately under this folder:
print ( next( os.walk(dir) )[2] )

#List all folders immediately under this folder:
print ( next( os.walk(dir) )[1] )

Python 3.5以降では、のos.scandir代わりに使用できますos.listdir。文字列の代わりに、DirEntryオブジェクトのイテレータを返します。ドキュメントから：

のscandir()代わりにを使用すると、listdir()ファイルタイプまたはファイル属性情報も必要とするコードのパフォーマンスが大幅に向上しますDirEntry。これは、ディレクトリのスキャン時にオペレーティングシステムが情報を提供すると、オブジェクトがこの情報を公開するためです。すべてのDirEntryメソッドがシステムコールを実行できますがis_dir()、is_file()通常はシンボリックリンクのシステムコールのみが必要です。DirEntry.stat()Unixでは常にシステムコールが必要ですが、Windowsではシンボリックリンクに1つだけ必要です。

あなたは経由して、オブジェクトの名前にアクセスすることができDirEntry.name、その後の出力に相当しますos.listdir

次のこともできます。

for path, subdirs, files in os.walk(dir_name):
    for name in files:
        if path == ".": #this will filter the files in the current directory
             #code here

これは私がそれを解決した方法です

if recursive:
    items = os.walk(target_directory)
else:
    items = [next(os.walk(target_directory))]

...

listdirを使用すると問題が発生します。os.path.isdir（identifier）は絶対パスでなければなりません。あなたがするサブディレクトリを選ぶには：

for dirname in os.listdir(rootdir):
  if os.path.isdir(os.path.join(rootdir, dirname)):
     print("I got a subdirectory: %s" % dirname)

代替方法は、os.path.join（）なしでテストを行うためにディレクトリに変更することです。

このスニペットを使用できます

for root, dirs, files in os.walk(directory):
    if level > 0:
        # do some stuff
    else:
        break
    level-=1

除外リストを作成し、fnmatchを使用してディレクトリ構造をスキップしてプロセスを実行します

excludes= ['a\*\b', 'c\d\e']
for root, directories, files in os.walk('Start_Folder'):
    if not any(fnmatch.fnmatch(nf_root, pattern) for pattern in excludes):
        for root, directories, files in os.walk(nf_root):
            ....
            do the process
            ....

'includes'と同じ：

if **any**(fnmatch.fnmatch(nf_root, pattern) for pattern in **includes**):

単純にa rangeを使用os.walkしてzip？と組み合わせないのはなぜですか？最善の解決策ではありませんが、うまくいくでしょう。

たとえば、次のようになります。

# your part before
for count, (root, dirs, files) in zip(range(0, 1), os.walk(dir_name)):
    # logic stuff
# your later part

Python 3で動作します。

また、A breakも単純です。（@Pieterからの回答を見てください）

アレックスの答えを少し変更しましたが、以下を使用してい__next__()ます：

print(next(os.walk('d:/'))[2]) または print(os.walk('d:/').__next__()[2])

[2]ことfileでroot, dirs, file他の回答で述べました

os.walkが検出するすべてのディレクトリのルートフォルダが変更されます。root ==ディレクトリかどうかを確認することで解決します

def _dir_list(self, dir_name, whitelist):
    outputList = []
    for root, dirs, files in os.walk(dir_name):
        if root == dir_name: #This only meet parent folder
            for f in files:
                if os.path.splitext(f)[1] in whitelist:
                    outputList.append(os.path.join(root, f))
                else:
                    self._email_to_("ignore")
    return outputList

import os

def listFiles(self, dir_name):
    names = []
    for root, directory, files in os.walk(dir_name):
        if root == dir_name:
            for name in files:
                names.append(name)
    return names