python以后缀名查找目录下的所有文件

遍历函数：

def file_gothrough(path,suffix):
    file_list=[]
    for root,dirs,files in os.walk(path):
        for file in files:
            if file[-3:]==suffix:
               file_list.append(os.path.join(root,file))
    return file_list

函数调用：

f=file_gothrough('data','txt')

即查找‘data’目录下的所有txt文件，返回一个列表f