在 python 中使用 OS.Walk() 計算子文件夾

Question

我想學習當 os.walk 進入另一個子文件夾時如何計數。 現在我能夠計算文件名中的文件數，但是當它從起始目錄切換到另一個子文件夾目錄時，我希望將一個計數添加到子文件夾中。 我不確定這樣做的最佳方法。 任何幫助是極大的贊賞。

count = 0 
subfolder_count = 0 

for foldername, subfolders, filenames in os.walk('articles'):
    for file in filenames:
        if file.endswith('.pdf'):
            count += 1

Answer 1

你已經在迭代了，繼續迭代！

count = 0
subfolder_count = 0

for root, dirs, files in os.walk('articles'):
    for dir_ in dirs:
        subfolder_count += 1
        for file_ in files:
            if file_.endswith('.pdf'):
                count += 1

在現代 Python (3.4+) 中，您甚至不需要os.walk ，因為pathlib已經涵蓋了兩次遍歷樹的小開銷。

import pathlib

path = pathlib.Path('articles')
subfolder_count = len(path.glob("**/"))
count = len(path.glob("**/*.pdf"))