如何獲取包含子目錄的目錄中的特定文件總數？

Question

以下Python代碼計算總的文件，我有一個包含多個子目錄的目錄的數量。結果打印子目錄名稱以及它包含的文件數量。

如何修改這個讓：

• 它只是尋找一個特定的文件擴展名（即「* .SHP」）
• 它提供的每個「.SHP」文件這兩個數子目錄和所有 「.SHP」 的最終計數文件

下面是代碼：

import os 
path = 'path/to/directory' 
folders = ([name for name in os.listdir(path)]) 
for folder in folders: 
    contents = os.listdir(os.path.join(path,folder)) 
    print(folder,len(contents)) 

Answer 1

可以在字符串上使用.endswith（）函數。這對於識別擴展很方便。你可以遍歷內容來找到這些文件，然後如下。

targets = [] 
for i in contents: 
    if i.endswith(extension): 
     targets.append(i) 
print(folder, len(contents)) 

Answer 2

感謝您的意見和答案，這是我使用的代碼（隨意標誌我的問題作爲鏈接的問題的副本，如果距離太近）：

import os 
path = 'path/to/directory' 
folders = ([name for name in os.listdir(path)]) 
targets = [] 
for folder in folders: 
    contents = os.listdir(os.path.join(path,folder)) 
    for i in contents: 
     if i.endswith('.shp'): 
      targets.append(i) 
    print(folder, len(contents)) 

print "Total number of files = " + str(len(targets))