python利用shp面对tif进行批量裁剪（掩膜）

import os
from osgeo import gdal, ogr
 
# 设置输入输出路径
input_shp_path = 'input_mask.shp'
input_tif_folder_path = 'input_tif_folder'
output_tif_folder_path = 'output_tif_folder'
 
# 检查输出文件夹是否存在，不存在则创建
if not os.path.exists(output_tif_folder_path):
    os.makedirs(output_tif_folder_path)
 
# 打开shapefile
shp_ds = ogr.Open(input_shp_path, 0)
shp_layer = shp_ds.GetLayer()
 
# 遍历shapefile中的每个要素
for feature in shp_layer:
    geom = feature.GetGeometryRef()
    
    # 打开tif文件
    tif_names = os.listdir(input_tif_folder_path)
    for tif_name in tif_names:
        input_tif_path = os.path.join(input_tif_folder_path, tif_name)
        output_tif_path = os.path.join(output_tif_folder_path, os.path.splitext(tif_name)[0] + '_masked.tif')
        
        # 读取raster数据
        src_ds = gdal.Open(input_tif_path, gdal.GA_ReadOnly)
        if src_ds is None:
            print(f'Could not open {input_tif_path}')
            continue
        
        # 创建masked raster
        mem_drv = gdal.GetDriverByName('GTiff')
        mem_ds = mem_drv.CreateCopy('', src_ds, 0, ['INTERLEAVE=PIXEL', 'COMPRESS=LZW'])
        
        # 应用掩膜
        gdal.SetConfigOption('GDAL_FILENAME_IS_UTF8', 'YES')
        gdal.RasterizeLayer(mem_ds, [1], shp_layer, options=['ATTRIBUTE=FID'])
        
        # 保存masked raster
        mem_ds.SetProjection(src_ds.GetProjection())
        mem_ds.SetGeoTransform(src_ds.GetGeoTransform())
        mem_ds = None
        
        # 重命名masked raster
        os.rename(output_tif_path.replace('.tif', '_tmp.tif'), output_tif_path)
 
# 关闭shapefile
shp_ds = None

这段代码使用了GDAL库来读取和处理GeoTiff数据，并且使用OGR库来读取Shapefile中的多边形掩膜。它会遍历一个文件夹内所有的GeoTiff文件，并将它们与Shapefile中的多边形进行匹配，然后将不属于多边形区域内的像素裁剪掉，形成一个新的masked GeoTiff文件。