Python实现 查找某文件夹中有多少个某种类型文件

起因是领导安排任务需要统计下载了多少景遥感图像，这些图像散落在某文件夹下不同级别的子文件夹中，一个一个找着数太费劲了，所以想到用python去统计一下，这些文件都是以.tif栅格影像格式，所以只要识别文件是否以.tif格式结尾就行了，代码如下：

import os  # 导入os模块，用于操作文件和目录
import re  # 导入re模块，用于正则表达式匹配

def count_files(directory):  # 定义函数count_tif_files，参数为directory
    count = 0  # 初始化count为0
    for root, dirs, files in os.walk(directory):  # 遍历directory目录下的所有文件和目录
        for file in files:  # 遍历当前目录下的所有文件
            if re.search(r'\.tif$', file):  # 如果文件名以.tif结尾
                ount += 1  # count加1
    return count  # 返回count

directory = 'F:\\Date'  # 设置directory为'F:\\Date'
count = count_files(directory)  # 调用count_tif_files函数，参数为directory，返回值赋给count
print(f'您的文件夹"{directory}"及其子文件夹中目标类型文件数量为 **{count}**。')  # 输出结果

大家使用的时候只要在文件夹中新建一个 随便名字.py，再把这串代码放进去

把中间if中的.tif改成你需要的.jpg或者.txt等等任何格式

把倒数第三行单引号部分改成你的目标文件夹路径

然后运行该文件就行，如果用powershell的话就是

cd (随便名字.py 所在的文件夹)

py 随便名字.py

或者直接简单点用VS Code右上角点击运行旁边小三角运行python文件，就可以输出正确结果了

如下图：