Đẹp hơn nhiều so với ở trên, vì bạn không cần vài os.path.join () và bạn sẽ nhận được đường dẫn đầy đủ trực tiếp (nếu bạn muốn), bạn có thể thực hiện điều này trong Python 3.5 trở lên.
subfolders = [ f.path for f in os.scandir(folder) if f.is_dir() ]
Điều này sẽ đưa ra đường dẫn đầy đủ đến thư mục con. Nếu bạn chỉ muốn tên của thư mục con sử dụng f.namethay vìf.path
https://docs.python.org/3/l Library / os.html # os.scandir
Hơi OT: Trong trường hợp bạn cần tất cả các thư mục con đệ quy và / hoặc tất cả các tệp đệ quy , hãy xem chức năng này, nhanh hơn os.walk& globvà sẽ trả về một danh sách tất cả các thư mục con cũng như tất cả các tệp bên trong các thư mục con (phụ) đó: https://stackoverflow.com/a/59803793/2441026
Trong trường hợp bạn chỉ muốn tất cả các thư mục con đệ quy :
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
Trả về một danh sách tất cả các thư mục con với đường dẫn đầy đủ của chúng. Điều này một lần nữa là nhanh hơn os.walkvà nhanh hơn rất nhiều glob.
Một phân tích của tất cả các chức năng
tl; dr:
- Nếu bạn muốn nhận tất cả các thư mục con ngay lập tức để sử dụng thư mục os.scandir.
- Nếu bạn muốn nhận tất cả các thư mục con, thậm chí các thư mục lồng nhau , hãy sử dụng os.walkhoặc - nhanh hơn một chút - fast_scandirchức năng trên.
- Không bao giờ chỉ sử dụng os.walkcho các thư mục con cấp cao nhất, vì nó có thể chậm hơn hàng trăm (!) Lần os.scandir.
- Nếu bạn chạy mã bên dưới, hãy đảm bảo chạy mã một lần để hệ điều hành của bạn sẽ truy cập vào thư mục, loại bỏ kết quả và chạy thử nghiệm, nếu không kết quả sẽ bị sai lệch.
- Bạn có thể muốn trộn lẫn các cuộc gọi chức năng, nhưng tôi đã thử nghiệm nó và nó không thực sự quan trọng.
- Tất cả các ví dụ sẽ cung cấp đường dẫn đầy đủ đến thư mục. Ví dụ pathlib là một đối tượng Đường dẫn (Windows).
- Phần tử đầu tiên
os.walksẽ là thư mục cơ sở. Vì vậy, bạn sẽ không chỉ nhận được thư mục con. Bạn có thể sử dụng fu.pop(0)để loại bỏ nó.
- Không có kết quả sẽ sử dụng phân loại tự nhiên . Điều này có nghĩa là kết quả sẽ được sắp xếp như sau: 1, 10, 2. Để có được sự sắp xếp tự nhiên (1, 2, 10), vui lòng xem tại https://stackoverflow.com/a/48030307/2441026
Kết quả :
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
Đã thử nghiệm với 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()