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文件。