Wie kann ich mit os.walk() einen visuell strukturierten Verzeichnisbaum in Python erstellen?

OS.walk() für rekursiven Verzeichnisdurchlauf in Python verwenden

Das rekursive Navigieren in Verzeichnissen und das Auflisten ihrer Inhalte ist eine häufige Aufgabe in der Python-Programmierung . Die Funktion os.walk() bietet eine effiziente Möglichkeit, dieses Ziel zu erreichen.

Originalcode

Der folgende Code zeigt, wie os.walk() zum Navigieren verwendet wird Vom Stammverzeichnis zu allen Unterverzeichnissen und drucken Sie deren Inhalte aus:

#!/usr/bin/python

import os
import fnmatch

for root, dir, files in os.walk("."):
        print(root)
        print("")
        for items in fnmatch.filter(files, "*"):
                print("..." + items)
        print("")

Dieser Code listet effektiv den Inhalt des aktuellen Verzeichnisses und seiner Unterverzeichnisse auf, tut dies jedoch nicht zwischen Verzeichnissen und Dateien unterscheiden.

Gewünschte Ausgabe

Die gewünschte Ausgabe erfordert jedoch die Anzeige von Verzeichnissen als Unterordner und Dateien als Elemente innerhalb dieser Ordner:

A
---a.txt
---b.txt
---B
------c.out

Überarbeiteter Code

Um die gewünschte Ausgabe zu erzielen, wurde Folgendes überarbeitet Code verwendet os.path.basename(), um den Verzeichnispfad zu erhalten:

#!/usr/bin/python

import os

# traverse root directory, and list directories as dirs and files as files
for root, dirs, files in os.walk("."):
    path = root.split(os.sep)
    print((len(path) - 1) * '---', os.path.basename(root))
    for file in files:
        print(len(path) * '---', file)

Dieser Code teilt zunächst den Pfad mithilfe von os.sep auf, um die Verzeichnisebenen zu identifizieren. Anschließend wird der Verzeichnisname mit einer Anzahl von Bindestrichen entsprechend seiner Tiefe gedruckt, gefolgt von einer Liste der Dateien in diesem Verzeichnis mit einem zusätzlichen Bindestrich für jede Ebene.

Das obige ist der detaillierte Inhalt vonWie kann ich mit os.walk() einen visuell strukturierten Verzeichnisbaum in Python erstellen?. Für weitere Informationen folgen Sie bitte anderen verwandten Artikeln auf der PHP chinesischen Website!