Python批量分割栅格文件为四等分或N×M个小块的方法

Python批量分割栅格文件为四等分或N×M个小块的方法

首页休闲益智超级疯狂切割更新时间:2024-05-11

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

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

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

  我们希望实现的是,将每一个子文件夹中的每一景遥感影像,都加以裁剪、切割,将原本的一景遥感影像切分为新的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)

  首先,我们需要设置输入和输出文件夹路径;其中,tif_file_path表示存放原始遥感影像的大文件夹路径,而result_file_path则表示存放拆分后结果的文件夹路径。随后,这里需要设置一下ArcPy的环境参数,通过arcpy.env.parallelProcessingFactor = 0这句代码禁用并行处理,以确保在处理期间不会产生问题——关于这一设置的原理,大家参考文章ArcGIS中ArcMap分割栅格Split Raster工具没有结果的解决(https://fkxxgis.blog.csdn.net/article/details/128474804)即可。

  随后,我们使用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基于遥感影像的文件名称将不同文件复制到对应的文件夹中(https://blog.csdn.net/zhebushibiaoshifu/article/details/129125339)、Python遍历多个子文件夹并基于文件名特征将文件复制到不同的目标文件夹(https://blog.csdn.net/zhebushibiaoshifu/article/details/132332068)中提到的代码加以自动实现。

  至此,大功告成。

欢迎关注:疯狂学习GIS

查看全文
大家还看了
也许喜欢
更多游戏

Copyright © 2024 妖气游戏网 www.17u1u.com All Rights Reserved