当前位置：首页 > article >正文

【ArcGIS】批量对栅格图像按要素掩膜提取

article 2024/9/24 5:29:02

要把一张大的栅格图裁成分省或者分县市的栅格集，一般是用ArcGIS里的按掩膜提取。
在这里插入图片描述
但是有的时候所要求的栅格集量非常大，所以用代码来做批量掩膜（按字段）会非常方便。

import arcpy , shutil , os
from arcpy import env
from arcpy.sa import *


#使用说明
print "开始使用前，请认真阅读使用说明"
print "\n"
print "使用说明 1 ：在工作空间内存放所需要掩膜提取的栅格文件。"
print "使用说明 2 ：按字段提取的要素一般是.shp文件，在输入路径时务必确保以.shp为结尾。"
print "使用说明 3 ：字段名是 使用说明 2 中要素里拥有的字段，一般使用名称作为输入要素。输入之前，务必确保与要素文件中字段名一致。"
print "使用说明 5 ：栅格须存放在工作空间内，在这里，只能输入一个栅格，所输入的栅格名不能加文件后缀。"
print "使用说明 4 ：输出文件夹作为存储掩膜后的栅格文件，文件夹里是一个栅格集。"
print "\n"


WorkSpace = raw_input("请输入工作空间路径:")
FieldFeature = raw_input("请输入按字段提取的要素路径：")
Field = raw_input("请输入要素字段名：")
Raster = raw_input("请输入需要裁剪的栅格名：")
OutputFile = raw_input("请输入输出文件夹路径：")
Free = raw_input("----------- 按回车开始执行 -----------")

print "\n"




if not os.path.exists(WorkSpace):
    print "工作空间不存在"
else:
    print "工作空间读取成功"
        
    
if not os.path.exists(OutputFile):
    print "输出路径不存在"
else:
    print "输出路径读取成功"



if os.path.exists(WorkSpace) and os.path.exists(OutputFile):         #当工作空间与临时文件夹都存在方能执行
    print "\n"
    print "正在检查程序运行环境，请稍等"
    print "\n"



    arcpy.env.workspace = WorkSpace                         #设置工作空间

    


    TempFile = WorkSpace + "/" + "TempFile"                   #设置临时文件夹路径
    

    if os.path.exists(TempFile):
        Enquire1 = raw_input("已存在临时文件夹TempFile，是否重建临时文件夹：")
        if Enquire1 == "是":
            shutil.rmtree(TempFile)
            print "临时文件夹删除成功"

                
    if not os.path.exists(TempFile):                         
        os.makedirs(TempFile)                                  #创建临时文件夹,分割输出文件夹，掩膜提取裁剪要素文件夹
        print "临时文件夹创建成功"                   
        print TempFile                                           #输出临时文件夹位置
        print "\n"


        print "正在执行程序，请稍等"
        print "\n"



                             

        #分割
        arcpy.Split_analysis(FieldFeature , FieldFeature , Field , TempFile)        #分割要素



        #按掩膜提取
        dirs = os.listdir(TempFile)                  
        for i in dirs:                           
            if os.path.splitext(i)[1] == ".shp":         #提取裁剪要素文件夹中.shp结尾的文件   分离path里的文件名和扩展名，文件名=[0],扩展名=[1]
                masks = TempFile + "/" + str(i)       #裁剪要素mask路径
                print str(i) + " 掩膜要素读取完毕"
                name = i.strip(".shp")                                                #去掉要素名称后缀.shp
                outExtractByMask = ExtractByMask(Raster, masks)                   #掩膜提取
                outpath = OutputFile + "/" + str(name)                           #掩膜提取输出路径
                outExtractByMask.save(outpath)                                  #保存输出文件
                print str(name) + " 掩膜提取执行完毕"
                print "\n"



        print "正在删除临时文件夹TempFile"
        print "\n"
        shutil.rmtree(TempFile)                                #强制删除临时文件夹
        print "临时文件夹TempFile已删除"
        print "\n"


            
        print "-----------按要素字段掩膜提取执行完毕-----------"





    else:
        print "已存在临时文件夹TempFile，无法执行"

else:
    print "执行未能成功"