如何获取包含子目录的目录中特定文件的总数？ [复制]

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

感谢 cmets 和回答，这是我使用的代码（如果我的问题太接近，请随时将我的问题标记为链接问题的副本）：

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