python将一个文件夹下的文件复制到另一个文件或是指定格式的文件

最新推荐文章于 2024-11-06 19:58:41 发布

愿望是当打工人

最新推荐文章于 2024-11-06 19:58:41 发布

阅读量2.8k

点赞数 2

分类专栏： YOLOv4 文章标签： python

转载注明出处。

本文链接：https://blog.csdn.net/weixin_40557160/article/details/112792825

版权

YOLOv4 专栏收录该内容

21 篇文章 2 订阅

订阅专栏

话不多说上代码：

将所有文件复制：

import shutil
import os

def remove_file(old_path, new_path):
    print(old_path)
    print(new_path)
    filelist = os.listdir(old_path) #列出该目录下的所有文件,listdir返回的文件列表是不包含路径的。
    print(filelist)
    for file in filelist:
        src = os.path.join(old_path, file)
        dst = os.path.join(new_path, file)
        print('src:', src)
        print('dst:', dst)
        shutil.move(src, dst)

if __name__ == '__main__':
    remove_file(r"/data/temp1", r"/data/temp2")

指定后缀的文件复制：

自己用的话简单修改下就可以，很简单。

import shutil
import os


def remove_file(old_path, new_path):
    # print(old_path)
    # print(new_path)
    if not os.path.exists(new_path):
        os.makedirs(new_path)
    filelist = os.listdir(old_path)  # 列出该目录下的所有文件,listdir返回的是文件列表是不包含路径的。
    # print(filelist)

    cout = 0
    for file in filelist:
        filename, extension = os.path.splitext(file)
        # if extension == ".png" or extension == ".jpg":
        if extension == ".json":
            print(file)
            cout += 1
            src = os.path.join(old_path, file)
            dst = os.path.join(new_path, file)
            # print('src:', src)
            # print('dst:', dst)
            shutil.copy(src, dst)

    print(cout)


if __name__ == '__main__':
    # remove_file("E:\\数据集\\total", "E:\\Desk\\JPEGImages")
    remove_file("E:\\数据集\\total", "E:\\Desk\\Annotation")