python 统计目录下文件匹配并计数

talk is cheap，show you the code：

import fnmatch
import os

def count_files(directory, prefix_list):
    lst = os.listdir(directory)
    cnt_list = [len(fnmatch.filter(lst, '*'+x)) for x in prefix_list]
    #前缀匹配
    #cnt_list = [len(fnmatch.filter(lst, x+'*')) for x in prefix_list]
    return cnt_list

res = count_files('./', ('.cpp', '.h', '.txt'))
print(res)

source code position: https://github.com/yjxiong/temporal-segment-networks/blob/master/pyActionRecog/utils/io.py