Python将名称中含有指定字段的文件批量复制

疯狂学习GIS

发布于 2024-08-20 20:44:45

830

发布于 2024-08-20 20:44:45

文章被收录于专栏：疯狂学习GIS

本文介绍基于Python语言，读取一个文件夹，并将其中每一个子文件夹内符合名称要求的文件加以筛选，并将筛选得到的文件复制到另一个目标文件夹中的方法。

本文的需求是：现在有一个大的文件夹，其中含有多个子文件夹，如下图所示。

对于其中的每一个子文件夹，都含有大量的文件；例如，我们就打开上图中的2021文件夹，得到结果如下图所示。

我们的需求是，希望基于每一个子文件夹内的每一个文件，将文件名称中含有指定字段的文件提取出来——在本文中，我们就希望提取得到文件名称中含有STB字段的，也就是上图紫色框内出现STB的文件是我们需要的。提取到全部符合要求的文件后，我们还需要将这些文件都复制到另一个目标文件夹中。

明确了需求，接下来就可以开始代码的撰写。本文所用到的代码如下所示。

#!/usr/bin/env python3
# -*- coding: utf-8 -*-
"""
Created on Mon Oct  2 12:48:49 2023

@author: fkxxgis
"""

import os
import shutil

source_folder = "/media/dell/H30-35/GF_SR"
target_folder = "/media/dell/H30-35/GF_SR/STB"

if not os.path.exists(target_folder):
    os.makedirs(target_folder)

year_folders = [f for f in os.listdir(source_folder) if os.path.isdir(os.path.join(source_folder, f))]

for year_folder in year_folders:
    if year_folder == "STB":
        continue
    year_folder_path = os.path.join(source_folder, year_folder)
    for file_name in os.listdir(year_folder_path):
        if "STB" in file_name:
            source_file = os.path.join(year_folder_path, file_name)
            target_file = os.path.join(target_folder, file_name)
            shutil.copy(source_file, target_file)

其中，我们导入了os和shutil模块，它们都提供了处理文件和目录的功能；同时，我们定义了源文件夹路径source_folder（也就是存储有多个子文件夹的文件夹路径）和目标文件夹路径target_folder（也就是最终结果存放的路径）。

随后，我们使用os.makedirs()函数检查目标文件夹是否存在，如果不存在则创建它。接下来，我们使用列表推导式生成一个包含源文件夹中所有子文件夹的列表year_folders；其中，我们使用了os.listdir()函数获取源文件夹中的所有文件和文件夹，并使用os.path.isdir()函数判断是否为文件夹。

接下来，我们就开始遍历每个子文件夹。首先，需要注意的是，我们要检查当前遍历的子文件夹是否为我们的目标文件夹路径STB；如果是，则跳过该子文件夹的处理——这一步骤是防止我们用来存放结果的目标文件夹也被遍历，否则就会出现文件复制冲突。

其次，构建当前遍历的子文件夹的完整路径year_folder_path，并使用os.listdir()函数获取子文件夹中的所有文件。随后，再遍历刚刚得到的子文件夹中的每个文件——如果文件名包含STB，那么这个文件就是我们需要的文件；我们就构建源文件（待复制的文件）的完整路径source_file和目标文件（预计复制完毕的文件）的完整路径target_file，并紧接着使用shutil.copy()函数将源文件复制到目标文件。