学术与工作效率

Python基于文件名将不同文件复制到不同路径下

2024-01-23  本文已影响0人  疯狂学习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)

  其中,我们导入了osshutil模块,它们都提供了处理文件和目录的功能;同时,我们定义了源文件夹路径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()函数将源文件复制到目标文件。

  通过这样的遍历,我们就能够将源文件夹路径source_folder中全部含有STB字段的文件复制到我们的目标文件夹中了。

  执行上述代码,即可在我们的目标文件夹中看到全部含有STB字段的文件,如下图所示。

  至此,大功告成。

上一篇下一篇

猜你喜欢

热点阅读