Python 获取文件夹下指定文件后缀数量

import os
from tkinter import filedialog

def GetDirFileCount(dirPath,suffix):
    count =0
    flist = os.listdir(dirPath)
    for i in flist:
        if os.path.splitext(i)[1] == '.{0}'.format(suffix):
            count +=1
    print('{0} 下格式为{1}的文件数量共有：{2}个'.format(dirPath,suffix,count))
    return count


foldPath = filedialog.askdirectory() # 获取选择的文件夹路径
GetDirFileCount(dirPath=foldPath,suffix='png')