Bagaimanakah saya boleh menggunakan os.walk() untuk mencipta pepohon direktori berstruktur visual dalam Python?

Menggunakan os.walk() untuk Recursive Directory Traversal dalam Python

Menavigasi direktori secara rekursif dan menyenaraikan kandungannya adalah tugas biasa dalam pengaturcaraan Python . Fungsi os.walk() menyediakan cara yang cekap untuk mencapai matlamat ini.

Kod Asal

Kod berikut menunjukkan cara menggunakan os.walk() untuk menavigasi daripada direktori akar kepada semua subdirektori dan cetak kandungannya:

#!/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("")

Kod ini menyenaraikan kandungan semasa dengan berkesan direktori dan subdirektorinya, tetapi ia tidak membezakan antara direktori dan fail.

Output yang Diingini

Walau bagaimanapun, output yang dikehendaki memerlukan paparan direktori sebagai subfolder dan fail sebagai item dalam folder tersebut:

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

Disemak Kod

Untuk mencapai output yang diingini, kod yang disemak berikut menggunakan os.path.basename() untuk mendapatkan laluan direktori:

#!/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)

Kod ini mula-mula memisahkan laluan menggunakan os.sep untuk mengenal pasti tahap direktori. Ia kemudian mencetak nama direktori dengan beberapa sempang sepadan dengan kedalamannya, diikuti dengan senarai fail dalam direktori tersebut dengan sempang tambahan untuk setiap peringkat.

Atas ialah kandungan terperinci Bagaimanakah saya boleh menggunakan os.walk() untuk mencipta pepohon direktori berstruktur visual dalam Python?. Untuk maklumat lanjut, sila ikut artikel berkaitan lain di laman web China PHP!