Bagaimanakah Saya Boleh Menyenaraikan Fail Sahaja Dengan Cekap dalam Direktori Menggunakan Python?

Mendapatkan Fail Dari Direktori

Mendapatkan semula senarai fail direktori ialah tugas biasa dalam pengaturcaraan Python. Satu cara untuk melakukan ini ialah menggunakan os.listdir(), yang mengembalikan semua item dalam direktori, termasuk kedua-dua fail dan direktori.

Pendekatan 1: Menggunakan os.path.isfile()

Untuk menyenaraikan fail sahaja, gunakan os.path.isfile():

from os import listdir
from os.path import isfile, join
mypath = 'path/to/directory'
onlyfiles = [f for f in listdir(mypath) if isfile(join(mypath, f))]

Pendekatan 2: Menggunakan os.walk()

os.walk() berulang pada direktori dan subdirektorinya , mengembalikan tuple untuk setiap direktori yang dilawatinya. Tuple ini mengandungi tiga senarai: satu untuk fail, satu untuk direktori dan satu untuk direktori semasa itu sendiri.

from os import walk
mypath = 'path/to/directory'
filenames = next(walk(mypath), (None, None, []))[2]  # [] if no files

Pendekatan ini menyediakan perwakilan struktur direktori yang lebih lengkap. Untuk menyenaraikan fail hanya dalam direktori semasa, putuskan gelung selepas lelaran pertama:

from os import walk

f = []
for (dirpath, dirnames, filenames) in walk(mypath):
    f.extend(filenames)
    break

Atas ialah kandungan terperinci Bagaimanakah Saya Boleh Menyenaraikan Fail Sahaja Dengan Cekap dalam Direktori Menggunakan Python?. Untuk maklumat lanjut, sila ikut artikel berkaitan lain di laman web China PHP!