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

最新推荐文章于 2024-10-19 20:58:09 发布

原创最新推荐文章于 2024-10-19 20:58:09 发布 · 951 阅读

2 ·

CC 4.0 BY-SA版权

文章标签：

#python #开发语言 #后端

此篇博客介绍了如何使用Python从指定目录中读取特定类型的文件（如.txt），如电影评论数据，通过os模块操作文件路径，并将文件内容复制到新的文件夹。重点在于数据预处理步骤和文件操作技巧。

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)