正则表达式选择特定文件进行复制

最新推荐文章于 2021-12-03 14:30:42 发布

Jia_11

最新推荐文章于 2021-12-03 14:30:42 发布

阅读量2.1k

点赞数 2

CC 4.0 BY-SA版权

分类专栏： python 文章标签：复制文件正则表达式特定文件名

本文链接：https://blog.youkuaiyun.com/u010103202/article/details/82858611

python 专栏收录该内容

25 篇文章

订阅专栏

本文介绍如何使用正则表达式筛选特定文件，并利用Python的shutil库进行文件复制。通过实例展示了从源目录复制符合特定模式的文件到目标目录的过程。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

正则表达式选择特定文件进行复制

文件复制

使用shutil库中的copy函数进行复制。

def copy_file(from_dir, to_dir, Name_list):
    if not os.path.isdir(to_dir):
        os.mkdir(to_dir)

    for name in Name_list:
        try:
            # print(name)
            if not os.path.isfile(os.path.join(from_dir, name)):
                print("{} is not existed".format(os.path.join(from_dir, name)))
            shutil.copy(os.path.join(from_dir, name), os.path.join(to_dir, name))
            # print("{} has copied to {}".format(os.path.join(from_dir, name), os.path.join(to_dir, name)))
        except:
            # print("failed to move {}".format(from_dir + name))
            pass
        # shutil.copyfile(fileDir+name, tarDir+name)
    print("{} has copied to {}".format(from_dir, to_dir))

正则表达式得到文件名列表

使用re库实现字符串匹配得到用户想要的文件夹名列表。参数pattern决定匹配file_list中哪些文件名。

def obtain_certrain_name_list(file_list, pattern):

    result = [re.findall(pattern, file) for file in file_list]
    flattened = sum(result, [])
    post_result = list(set(flattened))
    return post_result

总体代码示例

import re
import os
import random
import shutil


def copy_file(from_dir, to_dir, Name_list):
    if not os.path.isdir(to_dir):
        os.mkdir(to_dir)
    # 1
    # name_list = os.listdir(from_dir)

    # # 2
    # sample = random.sample(pathDir, 2)
    # print(sample)

    # 3
    for name in Name_list:
        try:
            # print(name)
            if not os.path.isfile(os.path.join(from_dir, name)):
                print("{} is not existed".format(os.path.join(from_dir, name)))
            shutil.copy(os.path.join(from_dir, name), os.path.join(to_dir, name))
            # print("{} has copied to {}".format(os.path.join(from_dir, name), os.path.join(to_dir, name)))
        except:
            # print("failed to move {}".format(from_dir + name))
            pass
        # shutil.copyfile(fileDir+name, tarDir+name)
    print("{} has copied to {}".format(from_dir, to_dir))


def obtain_certrain_name_list(file_list, pattern):

    result = [re.findall(pattern, file) for file in file_list]
    flattened = sum(result, [])
    post_result = list(set(flattened))
    return post_result


if __name__ == '__main__':

    from_path = "./Augment_from_files/"
    to_path = "./Augment_to_files/"
    if not os.path.exists(to_path):
        os.mkdir(to_path)

    from_file_list = os.listdir(from_path)
    # part1: copy gt image from from_path = "./Augment_from_files/"
    gt_image_list = obtain_certrain_name_list(from_file_list, ".+_gt.png")
    print(gt_image_list)
    gt_to_path = to_path + "annos"
    if not os.path.exists(gt_to_path):
        os.mkdir(gt_to_path)
    copy_file(from_path, gt_to_path, gt_image_list)

    # part2: copy image from from_path = "./Augment_from_files/"
    image_list = obtain_certrain_name_list(from_file_list, ".+[^_show].jpg")
    print(image_list)
    image_to_path = to_path + "images"
    if not os.path.exists(image_to_path):
        os.mkdir(image_to_path)
    copy_file(from_path, image_to_path, image_list)