Jauh lebih bagus daripada yang di atas, karena Anda tidak perlu beberapa os.path.join () dan Anda akan mendapatkan path lengkap secara langsung (jika Anda mau), Anda dapat melakukan ini dengan Python 3.5 dan lebih tinggi.
subfolders = [ f.path for f in os.scandir(folder) if f.is_dir() ]
Ini akan memberikan path lengkap ke subdirektori. Jika Anda hanya ingin menggunakan nama subdirektori f.namedaripadaf.path
https://docs.python.org/3/library/os.html#os.scandir
Sedikit Sl: Jika Anda membutuhkan semua subfolder secara rekursif dan / atau semua file secara rekursif , lihat fungsi ini, yang lebih cepat dari os.walk& globdan akan mengembalikan daftar semua subfolder serta semua file di dalam subfolder (sub-): https://stackoverflow.com/a/59803793/2441026
Jika Anda hanya menginginkan semua subfolder secara rekursif :
def fast_scandir(dirname):
subfolders= [f.path for f in os.scandir(dirname) if f.is_dir()]
for dirname in list(subfolders):
subfolders.extend(fast_scandir(dirname))
return subfolders
Mengembalikan daftar semua subfolder dengan jalur lengkapnya. Ini lagi lebih cepat daripada os.walkdan jauh lebih cepat daripada glob.
Analisis semua fungsi
tl; dr:
- Jika Anda ingin mendapatkan semua subdirektori langsung untuk penggunaan folder os.scandir.
- Jika Anda ingin mendapatkan semua subdirektori, bahkan yang bersarang , gunakan os.walkatau - sedikit lebih cepat - fast_scandirfungsi di atas.
- Jangan pernah gunakan os.walkhanya untuk subdirektori tingkat atas, karena bisa jadi ratusan (!) Kali lebih lambat daripada os.scandir.
- Jika Anda menjalankan kode di bawah ini, pastikan untuk menjalankannya sekali sehingga OS Anda akan mengakses folder, membuang hasil dan menjalankan tes, jika tidak hasilnya akan kacau.
- Anda mungkin ingin mencampur panggilan fungsi, tapi saya mengujinya, dan itu tidak masalah.
- Semua contoh akan memberikan path lengkap ke folder. Contoh pathlib sebagai objek Path (Windows).
- Elemen pertama
os.walkakan menjadi folder dasar. Jadi, Anda tidak hanya akan mendapatkan subdirektori. Anda dapat menggunakannya fu.pop(0)untuk menghapusnya.
- Tidak ada hasil yang akan menggunakan penyortiran alami . Ini berarti hasil akan diurutkan seperti ini: 1, 10, 2. Untuk mendapatkan penyortiran alami (1, 2, 10), silakan lihat di https://stackoverflow.com/a/48030307/2441026
Hasil :
os.scandir took 1 ms. Found dirs: 439
os.walk took 463 ms. Found dirs: 441 -> it found the nested one + base folder.
glob.glob took 20 ms. Found dirs: 439
pathlib.iterdir took 18 ms. Found dirs: 439
os.listdir took 18 ms. Found dirs: 439
Diuji dengan W7x64, Python 3.8.1.
# -*- coding: utf-8 -*-
# Python 3
import time
import os
from glob import glob
from pathlib import Path
directory = r"<insert_folder>"
RUNS = 1
def run_os_walk():
a = time.time_ns()
for i in range(RUNS):
fu = [x[0] for x in os.walk(directory)]
print(f"os.walk\t\t\ttook {(time.time_ns() - a) / 1000 / 1000 / RUNS:.0f} ms. Found dirs: {len(fu)}")
def run_glob():
a = time.time_ns()
for i in range(RUNS):
fu = glob(directory + "/*/")
print(f"glob.glob\t\ttook {(time.time_ns() - a) / 1000 / 1000 / RUNS:.0f} ms. Found dirs: {len(fu)}")
def run_pathlib_iterdir():
a = time.time_ns()
for i in range(RUNS):
dirname = Path(directory)
fu = [f for f in dirname.iterdir() if f.is_dir()]
print(f"pathlib.iterdir\ttook {(time.time_ns() - a) / 1000 / 1000 / RUNS:.0f} ms. Found dirs: {len(fu)}")
def run_os_listdir():
a = time.time_ns()
for i in range(RUNS):
dirname = Path(directory)
fu = [os.path.join(directory, o) for o in os.listdir(directory) if os.path.isdir(os.path.join(directory, o))]
print(f"os.listdir\t\ttook {(time.time_ns() - a) / 1000 / 1000 / RUNS:.0f} ms. Found dirs: {len(fu)}")
def run_os_scandir():
a = time.time_ns()
for i in range(RUNS):
fu = [f.path for f in os.scandir(directory) if f.is_dir()]
print(f"os.scandir\t\ttook {(time.time_ns() - a) / 1000 / 1000 / RUNS:.0f} ms.\tFound dirs: {len(fu)}")
if __name__ == '__main__':
run_os_scandir()
run_os_walk()
run_glob()
run_pathlib_iterdir()
run_os_listdir()