python 多文件夹中按条件复制到新的文件夹并重命名

而又何羡乎

已于 2022-02-12 23:38:37 修改

阅读量1.4k

点赞数

分类专栏： python文件处理文章标签： python 文件处理

于 2022-02-12 23:37:11 首次发布

本文链接：https://blog.csdn.net/qq_44285092/article/details/122903946

版权

python文件处理专栏收录该内容

2 篇文章 0 订阅

订阅专栏

这段代码定义了几个函数，包括读取文件路径、创建文件夹、复制并重命名文件。主要功能是从2010、2011和label文件夹中读取文件，将它们复制到新的train1文件夹下，并根据文件类型进行重命名。标签系统未明确，但可能与文件处理或数据预处理相关。

摘要由CSDN通过智能技术生成

在这里插入图片描述

将文件夹2010,2011，label中的文件，复制到res/train1 … 文件中去，train1 是新建的文件夹，并重命名

import shutil
import os

def read_file_path(path):
    # 读取文件名
    return [i for i in os.listdir(path)]

def mkdir(path):
    # 创建文件夹
    import os
    path = path.strip()
    path = path.rstrip("\\")
    isExists = os.path.exists(path)
    if not isExists:
        os.makedirs(path)
        # print (path+' 创建成功')
        return True
    else:
        # print(path+' 目录已存在')
        return False

def copy_file_rename(old_path, new_path):
    # 复制文件到指定地址
    try:
        shutil.copy(old_path, new_path)
    except Exception as e:
        print(e)

def res(path_list):
    label = 0
    for i in path_list:
        file_list = read_file_path(i)
        for j in file_list:
            old_path = f"{i}/{j}"
            mkdir(r'res/train{}'.format(j.split(".")[0]))
            if label == 0:
                new_path = r'res/train{}'.format(j.split(".")[0]) + "/t0.tif"
            elif label == 1:
                new_path = r'res/train{}'.format(j.split(".")[0]) + "/t1.tif"
            else:
                new_path = r'res/train{}'.format(j.split(".")[0]) + "/label.png"
            copy_file_rename(old_path, new_path)
        label += 1

# 顺序别放错了，不然就出错了
res(["2010", "2011", "label"])