os.walk ohne in die folgenden Verzeichnisse zu graben

Wie beschränke ich mich darauf os.walk, nur Dateien in dem von mir bereitgestellten Verzeichnis zurückzugeben?

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

Verwenden Sie die walklevelFunktion.

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

Es funktioniert genauso wie os.walk, aber Sie können ihm einen levelParameter übergeben, der angibt, wie tief die Rekursion gehen wird.

Verwenden Sie os.walk nicht.

Beispiel:

import os

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

Ich denke, die Lösung ist eigentlich sehr einfach.

verwenden

break

Um nur die erste Iteration der for-Schleife durchzuführen, muss es einen eleganteren Weg geben.

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

Wenn Sie os.walk zum ersten Mal aufrufen, werden Tulpen für das aktuelle Verzeichnis zurückgegeben und in der nächsten Schleife der Inhalt des nächsten Verzeichnisses.

Nehmen Sie das Original-Skript und fügen Sie einfach eine Pause hinzu .

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

Der Vorschlag listdirist gut. Die direkte Antwort auf Ihre Frage in Python 2 lautet root, dirs, files = os.walk(dir_name).next().

Die entsprechende Python 3-Syntax lautet root, dirs, files = next(os.walk(dir_name))

Sie können verwenden os.listdir(), um eine Liste von Namen (sowohl für Dateien als auch für Verzeichnisse) in einem bestimmten Verzeichnis zurückzugeben. Wenn Sie zwischen Dateien und Verzeichnissen unterscheiden müssen, rufen Sie os.stat()jeden Namen auf.

Wenn Sie komplexere Anforderungen als nur das oberste Verzeichnis haben (z. B. VCS-Verzeichnisse ignorieren usw.), können Sie auch die Liste der Verzeichnisse ändern, um zu verhindern, dass os.walk diese erneut durchläuft.

dh:

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

Hinweis - Achten Sie darauf, die Liste zu mutieren, anstatt sie nur erneut zu binden. Offensichtlich weiß os.walk nichts über die externe Rückbindung.

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

Die gleiche Idee mit listdir, aber kürzer:

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

Ich hatte das Gefühl, meine 2 Pence hineinzuwerfen.

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

In Python 3 konnte ich Folgendes tun:

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

Seit Python 3.5 können Sie os.scandiranstelle von verwenden os.listdir. Anstelle von Zeichenfolgen erhalten Sie DirEntryim Gegenzug einen Iterator von Objekten. Aus den Dokumenten:

Die Verwendung von scandir()anstelle von listdir()kann die Leistung von Code, der auch Dateityp- oder Dateiattributinformationen benötigt, erheblich steigern, da DirEntryObjekte diese Informationen verfügbar machen, wenn das Betriebssystem sie beim Scannen eines Verzeichnisses bereitstellt. Alle DirEntryMethoden können einen Systemaufruf durchführen, aber is_dir()und is_file()in der Regel nur einen Systemaufruf für symbolische Links erfordern; DirEntry.stat()erfordert unter Unix immer einen Systemaufruf, unter Windows jedoch nur einen für symbolische Links.

Sie können auf den Namen des Objekts zugreifen DirEntry.name, über den dann die Ausgabe von entsprichtos.listdir

Sie können auch Folgendes tun:

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

So habe ich es gelöst

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

...

Bei der Verwendung von listdir gibt es einen Haken. Der os.path.isdir (Bezeichner) muss ein absoluter Pfad sein. So wählen Sie Unterverzeichnisse aus:

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

Die Alternative besteht darin, in das Verzeichnis zu wechseln, um die Tests ohne os.path.join () durchzuführen.

Sie können dieses Snippet verwenden

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

Erstellen Sie eine Liste mit Ausschlüssen, überspringen Sie mit fnmatch die Verzeichnisstruktur und führen Sie den Vorgang aus

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
            ....

das gleiche wie für 'enthält':

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

Warum nicht einfach ein rangeund os.walkkombiniert mit dem zip? Ist nicht die beste Lösung, würde aber auch funktionieren.

Zum Beispiel so:

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

Funktioniert für mich auf Python 3.

Auch: A breakist übrigens auch einfacher. (Schauen Sie sich die Antwort von @Pieter an)

Eine kleine Änderung an Alex 'Antwort, aber mit __next__():

print(next(os.walk('d:/'))[2]) oder print(os.walk('d:/').__next__()[2])

mit dem [2]Wesen der filein root, dirs, filein anderen Antworten erwähnt

Änderungen am Stammordner für jedes Verzeichnis, das os.walk findet. Ich löse diese Überprüfung, ob root == Verzeichnis

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