python 获取文件夹下指定后缀的文件

import os
import sys
import os.path
def list_file(path):
    count = 0
    for filename in os.listdir(path):
        if os.path.splitext(filename)[1] == '.csv':
            count = count+1
            print(filename)
            #fp = open(dirname+os.sep+filename,'r')
            #print(len(fp.readlines())-1)
            #fp.close()
    print('total csv file: ',count)
if __name__=="__main__":
    list_file(r'F:\uci\data')