统计指定文件后缀的文件数量

import os


def file_amount(file_suffix: list, dir_path: str):
    """
    统计指定文件后缀的文件数量
    Args:
        file_suffix: list
            文件后缀, e.g. file_suffix=['jpg', 'png', 'bmp']
        dir_path: str
            文件目录
    Returns:
        num: int
            文件数量
    """
    num = 0
    for root, dirs, files in os.walk(dir_path):
        for file_name in files:
            # 跳过不满足后缀要求的文件
            if file_name.split('.')[-1] not in file_suffix:
                continue
            num += 1
    return num


if __name__ == '__main__':
    suffix = ['jpg', 'png', 'bmp']
    dirs = r"D:\datasets"
    amount = file_amount(suffix, dirs)
    print('文件数量:', amount)