基于ArcPy裁剪多个栅格图像为四等分(或N×M等分)
本文介绍基于Python中的ArcPy
模块,基于一个大文件夹,遍历其中每一个子文件夹中所有的遥感影像栅格文件,并将原本的每一景遥感影像文件四等分切割,或裁剪为其他指定个数的小块的方法。
首先明确一下我们的需求。现有一个大文件夹,其中具有多个子文件夹,如下图所示。
其中,每一个子文件夹中,都含有大量的栅格遥感影像文件(我们这里就以.tif
格式的栅格图像文件为例来介绍);例如,随意打开上图中任意一个子文件夹,都是如下图所示的情况。
我们希望实现的是,将每一个子文件夹中的每一景遥感影像,都加以裁剪、切割,将原本的一景遥感影像切分为新的4
份,相当于是四等分。这里大家想将原本的图像分为几部分都可以,即可以将原来的一景图像分为N * M
个部分。
了解了需求,我们就可以开始代码的撰写。本文所用代码,其实和我们之前的文章依据矢量图层批量将栅格图像切分为多个:Python ArcPy,以及Python批量切割遥感影像为网格状小区域的方法中提到的代码思路比较类似,但是在文件读取、裁剪参数设置等方面有所不同;大家如果有需要,可以先查阅以上这两篇文章。
本文所用代码如下。
# -*- 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
);如下图所示。
接下来,如果我们希望将这些得到的小遥感影像按照文件名的特征,复制、剪切到其他文件夹中,可以参考文章Python依据文件名称将大量文件复制到不同文件夹的方法,以及Python基于文件名将不同文件复制到不同路径下中提到的代码加以自动实现。
至此,大功告成。