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

话不多说上代码:

将所有文件复制:

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

 

    原文作者:愿望是当打工人
    原文地址: https://blog.csdn.net/weixin_40557160/article/details/112792825
    本文转自网络文章,转载此文章仅为分享知识,如有侵权,请联系博主进行删除。
点赞