python 返回指定目录下的所有文件与特定文件

有时要读取一个文件夹下的多个文件，下面这段代码即是返回指定文件目录下的所有文件的文件名，存于列表中，然后从列表中进行匹配文件名中含有'.csv'的文件名，最后返回所有'.csv'文件

读取'.csv'文件可以参考：

#!/usr/bin/env python
# -*- coding: UTF-8 -*-
import sys
import os
import re
def eachFile(filepath):
    pathDir =  os.listdir(filepath)  #返回指定路径下所有文件和文件夹的名字,并存放于一个列表中
    dir_list = []
    for allDir in pathDir:
        if re.search('.csv', allDir) != None:  #search()会扫描整个string查找匹配,会扫描整个字符串并返回第一个成功的匹配
            dir_list.append(allDir)
    return dir_list

if __name__ == '__main__':
    filepath = u"G:\\研究\\AWS\\AWS\\mod\\"  #指定文件目录
    dir_list = eachFile(filepath)  #获取文件名列表
    print(dir_list)