Как получить общее количество определенных файлов в каталоге, содержащем подкаталоги?

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))