在指定文件夹里找到文件名字相同的文件,并保存到指定的文件夹内

import os
import shutil


def copy_matching_files(
    source_dir, source_extension, target_dir, target_extension, destination_dir
):
    # 确保保存文件夹存在
    if not os.path.exists(destination_dir):
        os.makedirs(destination_dir)

    # 从A文件夹获取所有特定扩展名的文件的基本名称
    source_base_names = {
        os.path.splitext(f)[0]
        for f in os.listdir(source_dir)
        if f.endswith(source_extension)
    }

    # 从B文件夹获取所有特定扩展名的文件的完整路径
    target_files = {
        os.path.splitext(f)[0]: os.path.join(target_dir, f)
        for f in os.listdir(target_dir)
        if f.endswith(target_extension)
    }

    # 遍历A文件夹中的特定扩展名文件基本名称
    for base_name in source_base_names:
        # 检查B文件夹中是否存在具有相同基本名称的文件
        if base_name in target_files:
            # 获取文件的完整路径
            target_file_path = target_files[base_name]
            # 构建文件在目标文件夹中的新路径
            destination_file_path = os.path.join(
                destination_dir, os.path.basename(target_file_path)
            )
            # 复制文件
            shutil.copy2(target_file_path, destination_file_path)
            print(f"Copied: {os.path.basename(target_file_path)} to {destination_dir}")
        else:
            print(f"No corresponding file in B folder for base name: {base_name}")

    print("Copy process completed.")


# 使用示例
source_dir = "images/test"  # 源文件夹路径
source_extension = ".jpg"  # 源文件的扩展名
target_dir = "VOCdevkit/Annotations"  # 目标文件夹路径
target_extension = ".xml"  # 目标文件的扩展名
destination_dir = "xml/test"  # 保存复制文件的文件夹路径

copy_matching_files(
    source_dir, source_extension, target_dir, target_extension, destination_dir
)

 46行,是参考文件夹的路径,

47行,看是啥文件,是.jpg还是其他

48行,就是要找的文件夹路径,

49行,看找的文件夹文件后缀是啥,这里是.xml

50行,保存指定的路径

  • 3
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值