Python+ArcPy多个矢量文件批量掩膜多个栅格数据_python批量掩膜提取

在使用模型构建器等进行掩膜提取时，有时候会有要使用多个矢量文件去掩膜裁剪多个栅格数据的情况，模型构建器内只能进行一次迭代，无法直接进行多次迭代，在模型里面嵌套模型多次迭代还不如直接使用arcpy方便。

下面我有20来个矢量文件，要分别掩膜2000-2021年的栅格文件，基于arcpy

# coding=utf-8
# -*- coding:gbk-*-
import arcpy
import os

arcpy.env.overwriteOutput = True

# 定义函数，获取特定路径下特定后缀文件形成列表
def getFileName(Path, suffix):
    # 获取指定目录下的所有指定后缀的文件名
    input_template_All = []
    f_list = os.listdir(Path)  # 返回文件名
    for i in f_list:
        # os.path.splitext():分离文件名与扩展名
        if os.path.splitext(i)[1] == suffix:
            input_template_All.append(i)
            # print(i)
    return input_template_All


#设置工作空间，直接读取栅格数据
with arcpy.EnvManager(workspace=r'XXXX'):
    Tif_List = arcpy.ListRasters("*", "tif")
    print(Tif_List)
    ShpPath = "XXXX"
    Shp_List = getFileName(ShpPath, ".shp")
    print(Shp_List)
    #用于设置输出文件名
    TifName = 2000
    for tif in Tif_List:
        ShpName = 1
        for shp in Shp_List:
            if ShpName < 10:
                OutName = "XXXX"+str(TifName)+"sta0"+str(ShpName)+".tif"
            else:
                OutName = "XXXXX" + str(TifName) + "sta" + str(ShpName)+".tif"
            inshp = ShpPath +"/"+ shp
            OutTif = arcpy.sa.ExtractByMask(tif, inshp)
            OutTif.save(OutName)
            ShpName += 1
        TifName += 1
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41