likes
comments
collection
share

Python中ArcPy切割栅格文件为四等分

作者站长头像
站长
· 阅读数 11

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

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

Python中ArcPy切割栅格文件为四等分

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

Python中ArcPy切割栅格文件为四等分

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

  本文所用代码如下。

# -*- 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)

  随后,我们使用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开始计算,本文中就是0123);如下图所示。

Python中ArcPy切割栅格文件为四等分

  至此,大功告成。