ArcPy批量切割不同文件夹中栅格数据：逐一等分为若干小块

疯狂学习GIS

发布于 2024-07-22 17:18:24

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

本文介绍基于Python中的ArcPy模块，基于一个大文件夹，遍历其中每一个子文件夹中所有的遥感影像栅格文件，并将原本的每一景遥感影像文件四等分切割，或裁剪为其他指定个数的小块的方法。

首先明确一下我们的需求。现有一个大文件夹，其中具有多个子文件夹，如下图所示。

其中，每一个子文件夹中，都含有大量的栅格遥感影像文件（我们这里就以.tif格式的栅格图像文件为例来介绍）；例如，随意打开上图中任意一个子文件夹，都是如下图所示的情况。

我们希望实现的是，将每一个子文件夹中的每一景遥感影像，都加以裁剪、切割，将原本的一景遥感影像切分为新的4份，相当于是四等分。这里大家想将原本的图像分为几部分都可以，即可以将原来的一景图像分为N * M个部分。

了解了需求，我们就可以开始代码的撰写。本文所用代码，其实和我们之前的文章依据矢量要素批量分割栅格为多部分：ArcPy，以及ArcPy切分大量遥感影像为多个网格区域的方法中提到的代码思路比较类似，但是在文件读取、裁剪参数设置等方面有所不同；大家如果有需要，可以先查阅以上这两篇文章。

本文所用代码如下。

# -*- coding: utf-8 -*-
"""
Created on Mon Aug 28 22:39:52 2023

@author: fkxxgis
"""

import os
import arcpy

tif_file_path = r"E:/02_Project/202307_NDVIProduce/Beijing_Preprocessing/Original"
result_file_path = r"E:/02_Project/202307_NDVIProduce/Beijing_Preprocessing/Four_Result"
arcpy.env.parallelProcessingFactor = 0

for root, dirs, files in os.walk(tif_file_path):
    for dir_name in dirs:
        dir_path = os.path.join(root, dir_name)
        arcpy.env.workspace = dir_path
        tif_file_list = arcpy.ListRasters("*", "tif")

        for tif_file in tif_file_list:
            arcpy.SplitRaster_management(tif_file,
                                         result_file_path,
                                         tif_file.split(".tif")[0] + "_",
                                         "NUMBER_OF_TILES",
                                         "TIFF",
                                         "BILINEAR",
                                         "2 2",
                                         "#",
                                         "#",
                                         "PIXELS",
                                         "#",
                                         "#",
                                         "#",
                                         "#",
                                         "#",
                                         "0")
        print(dir_path)

首先，我们需要设置输入和输出文件夹路径；其中，tif_file_path表示存放原始遥感影像的大文件夹路径，而result_file_path则表示存放拆分后结果的文件夹路径。随后，这里需要设置一下ArcPy的环境参数，通过arcpy.env.parallelProcessingFactor = 0这句代码禁用并行处理，以确保在处理期间不会产生问题——关于这一设置的原理，大家参考文章ArcMap栅格分割工具Split Raster得不到结果怎么办？即可。

随后，我们使用os.walk，遍历源文件夹中的所有子文件夹；对于每个子文件夹，设置ArcPy工作环境为该子文件夹的路径，从而使得我们可以使用 arcpy.ListRasters获取这一个子文件夹中的所有.tif格式的遥感影像文件。

接下来，对于每个遥感影像，使用arcpy.SplitRaster_management函数对其进行切分。其中，第一个参数tif_file是要拆分的遥感影像的路径，第二个参数result_file_path是拆分后结果保存的文件夹路径，第三个参数tif_file.split(".tif")[0] + "_"是输出文件的前缀，这里使用原始文件名去掉.tif后缀，并在末尾添加下划线；接下来的参数用于设置拆分方式、输出格式、插值方法等。这里我们选择将遥感影像拆分为2 * 2块（大家如果需要将遥感影像拆分为其他的数量，修改这里就可以），插值方法为BILINEAR，输出格式为TIFF。最后一个参数0表示将原本遥感影像中的0值作为切割后小图像的NoData值。

运行上述代码，我们即可在结果文件夹中，看到每一景遥感影像都已经被切分为了4份，且这里每一份的文件后都带有一个数字后缀（数字后缀从0开始计算，本文中就是0、1、2与3）；如下图所示。