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

深碍√TFBOYSˉ_ 2024-02-18 12:51 5阅读 0赞

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

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 "执行未能成功"

直接复制到IDLE里运行即可，使用说明写得很详细。

[71cff048b4a74ce698e0bbd7fe0d180b.png]: https://image.dandelioncloud.cn/pgy_files/images/2024/02/18/60862a7456714a91a0c9fccb986f0ca3.png