Python 搜索文件夹指定后缀文件

包括所有子文件夹

def filter_file(path,key):
    dic_list = os.walk(path)
    find_list = []
    for i in dic_list:
        if os.path.isdir(i[0]):
            header = i[0]
            file = i[2]
            for f in file:
                if f.rfind(key) != -1:
                    find_list.append([header,f])
    return find_list