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

Question

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

Answer 1

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 パラメーターを渡すことができます。

Answer 2

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

例：

import os

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

Answer 3

実際に解決策は非常に簡単だと思います。

使用

break

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

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

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

元のスクリプトを取得して、単に break を追加します。

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

Answer 4

listdir を使用することをお勧めします。 Python 2でのあなたの質問に対する直接的な答えは root、dirs、files = os.walk（dir_name）.next（）です。

同等のPython 3構文は、 root、dirs、files = next（os.walk（dir_name））

です。

Answer 5

os.listdir（ ） は、指定されたディレクトリ内の名前のリスト（ファイルとディレクトリの両方）を返します。ファイルとディレクトリを区別する必要がある場合は、名前ごとに os.stat（）を呼び出します。

Answer 6

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

ie：

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は外部の再バインドについて知りません。

Answer 7

listdir と同じ考え方ですが、短い：

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

Answer 8

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

Answer 9

私の2ペンスを投げ入れるような感じ。

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

Answer 10

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

Answer 11

次のこともできます。

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

Answer 12

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

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

...

Answer 13

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（）なしでテストを実行するディレクトリに変更することです。

Answer 14

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

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

Answer 15

除外リストを作成し、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**):

Answer 16

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

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

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

Python 3で動作します。

また： break は単純すぎます。 （@Pieterからの回答をご覧ください）

Answer 17

Alexの答えをわずかに変更しましたが、 __ next __（）を使用します：

print（next（os.walk（ 'd：/'））[2]） または     print（os.walk（ 'd：/'）.__ next __（）[2]）

他の回答で言及されている root、dirs、file の [2] が file である場合

Answer 18

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

  

listdir（）の代わりに scandir（）を使用すると、 DirEntry オブジェクトはこの情報を公開します。すべての DirEntry メソッドはシステムコールを実行できますが、 is_dir（）および is_file（）は通常、シンボリックリンクのシステムコールのみを必要とします。 DirEntry.stat（）は、Unixでは常にシステムコールを必要としますが、Windowsではシンボリックリンクに対してのみシステムコールを必要とします。

DirEntry.name を介してオブジェクトの名前にアクセスできます。これは、 os.listdir

の出力と同等です。

Answer 19

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