Bagaimanakah Saya Boleh Menyesuaikan Output Fungsi os.walk() Python untuk Mencipta Penyenaraian Direktori Bersarang?

Menggunakan os.walk() untuk Melintasi Direktori Secara Rekursif dalam Python: Panduan Terperinci

Pengenalan

Fungsi os.walk() Python menyediakan mekanisme yang berkuasa untuk melintasi direktori secara rekursif pokok. Ia melelar ke atas semua direktori dan fail dalam direktori tertentu, menjadikannya alat penting untuk tugas seperti pengurusan fail dan penerokaan direktori.

Masalah: Menyesuaikan Output Penyenaraian Direktori

Untuk menunjukkan keupayaan os.walk(), pertimbangkan senario berikut: anda mahu menavigasi secara rekursif daripada direktori akar dan mencetak senarai direktori dan fail tersuai, termasuk tahap bersarang.

Kod Permulaan dan O/P

Menggunakan kod berikut:

import os
import fnmatch

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

Anda mendapat output berikut:

.

...Python_Notes
...pypy.py
...pypy.py.save
...classdemo.py
....goutputstream-J9ZUXW
...latest.py
...pack.py
...classdemo.pyc
...Python_Notes~
...module-demo.py
...filetype.py

./packagedemo

...classdemo.py
...__init__.pyc
...__init__.py
...classdemo.pyc

Walau bagaimanapun, output ini tidak memenuhi format yang diingini daripada:

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

Penyelesaian

Untuk menyesuaikan output, pendekatan yang lebih baik diperlukan. Kod berikut mencapai format yang diingini:

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)

Penjelasan

Kod ini menggunakan logik berikut:

1. Ia membelah akar laluan ke laluan senarai menggunakan pemisah direktori sistem pengendalian (cth., '/' on UNIX).
2. Ia mencetak nama asas direktori semasa dengan bilangan sempang utama yang sesuai, mewakili tahap bersarangnya.
3. Ia melelang ke atas fail dalam direktori semasa dan mencetak setiap fail dengan tahap bersarangnya.

Output

Menggunakan ini kod yang dipertingkatkan, anda akan mendapat output yang diingini:

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

Dalam output ini, A dan B mewakili direktori, manakala a.txt, b.txt dan c.out mewakili fail. Bilangan awalan sempang menunjukkan tahap bersarang setiap item.

Atas ialah kandungan terperinci Bagaimanakah Saya Boleh Menyesuaikan Output Fungsi os.walk() Python untuk Mencipta Penyenaraian Direktori Bersarang?. Untuk maklumat lanjut, sila ikut artikel berkaitan lain di laman web China PHP!