Python识别文件名中的字段从而分类、归档栅格文件到不同文件夹

  本文介绍基于Python语言，针对一个文件夹下的大量栅格遥感影像文件，基于其各自的文件名，分别创建指定名称的新文件夹，并将对应的栅格遥感影像文件复制到不同的新文件夹下的方法。

  首先，我们来看一下本文需要实现的需求。现有一个文件夹，其中有大量.tif格式的栅格遥感影像文件，以及.xml、ovr等格式的遥感影像辅助信息文件，如下图所示。

  其中，如上图中紫色框所示，每一景遥感影像文件的文件名称中，都有一个表示其编号的字段；我们希望基于这一编号字段，将带有相同编号字段的栅格遥感影像文件，以及其对应的辅助信息文件，都复制到一个结果文件夹中；这个结果文件夹如下图所示。

  其中，结果文件夹内含有多个不同编号的子文件夹，这个编号就是上上图中，栅格遥感影像所带有的编号。例如，我们希望将所有文件名称中带有15字段的栅格遥感影像文件及其辅助信息文件，都复制到结果文件夹中名称为15的子文件夹中，以此类推。

  知道了具体需求，我们即可开始代码的撰写。本文所用到的代码如下所示。

1# -*- coding: utf-8 -*-

2"""

3Created on Mon Feb 13 20:25:04 2023

4

5@author: fkxxgis

6"""

7

8import os

9import re

10import shutil

11

12tif_file_path = r"E:\02_Project\01_Chlorophyll\Fishnet\Result"

13result_file_path = r"E:\02_Project\01_Chlorophyll\Fishnet\ResultFolder"

14

15file_list = os.listdir(tif_file_path)

16for file in file_list:

17    file_id_suffix = file.split("_")[2]

18    file_id = re.findall("\d+", file_id_suffix)[0]

19    if not os.path.exists(os.path.join(result_file_path, file_id)):

20        os.makedirs(os.path.join(result_file_path, file_id))

21        print("Make " + file_id + " path.")

22    shutil.copy(os.path.join(tif_file_path, file), os.path.join(result_file_path, file_id, file))

  其中，tif_file_path是存储有原有待复制遥感影像文件的文件夹，result_file_path则表示最终的结果文件夹。

  代码的整体思路也非常简单。首先，我们需要导入os、re与shutil等3个Python库，分别实现系统文件遍历、字符串匹配与遥感影像文件的复制。其次，我们通过os.listdir()函数，遍历待复制遥感影像文件对应的文件夹，获取其中的每一个文件；这里的文件既包括.tif格式的栅格遥感影像文件，同时也包括.xml、ovr等格式的遥感影像辅助信息文件。

  随后，对每一个文件加以处理。我们基于每一个文件的文件名称的规则，通过split()函数，将其中表示编号的字段以及这一字段之后的内容提取出来；紧接着，基于re.findall()函数，通过字符串匹配的方式，将表示编号的字段（也就是文件名称中的数字部分）提取出来。

  接下来，知道了当前文件对应的编号，我们就可以开始复制工作。但是这里需要注意，由于我们需要将每一个文件都放入结果文件夹中的子文件夹，因此需要首先判断当前子文件夹有没有被建立；如果没有创建的话，我们需要创建一下这个子文件夹。这一步骤通过if判断语句及其内部的代码即可实现——通过os.path.exists()函数判断是否存在指定的子文件夹，如果不存在的话就新建这一子文件夹。

  随后，就可以开始文件的复制工作了。这一步骤我们通过shutil.copy()函数即可实现。

  运行上述代码后，我们即可在结果文件夹中看到各个编号对应的子文件夹，并在子文件夹中看到这一编号对应的全部文件。如下图所示，可以看到结果文件夹中，名称为15的子文件夹内，包含的就是文件名称中带有15字段的所有遥感影像文件及其对应的辅助信息文件。

  至此，大功告成。

欢迎关注（几乎）全网：疯狂学习GIS