python根据文件的后缀名提取所需文件

def read_imdb(data_dir,filetype):
    """读取数据集文本序列和标签data_dir所要提取的文件夹名,所要提取的文件类型filetype（如txt，dll"""
    data = []
#提取到的文件所保存的文件夹名
    new_file='F:\\windll'
    for file in os.listdir(data_dir):
        #join用来连接文件路径
        filenamedir=os.path.join(data_dir, file)
        if filenamedir[-3:]==filetype:
            #上面的-3只适用于后缀名是三个字符的，其余情况自行更改
            new_file1=os.path.join(new_file,file)
            print(new_file1)
            shutil.copy(filenamedir, new_file1)