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')