使用Python和OGR/GDAL栅格化WKT面(来自geopandas)

import ogr
from osgeo import gdal
from shapely.wkt import loads

wkt = 'POLYGON ((0 0, 0 10, 10 10, 10 0, 0 0))'

geometry = loads(wkt)

pixel_size = 1  # 像素大小
x_min, x_max, y_min, y_max = geometry.bounds  # 获取几何对象的边界框
width = int((x_max - x_min) / pixel_size)  # 计算栅格宽度
height = int((y_max - y_min) / pixel_size)  # 计算栅格高度
target_ds = gdal.GetDriverByName('GTiff').Create('raster.tif', width, height, 1, gdal.GDT_Byte)  # 创建栅格数据集
target_ds.SetGeoTransform((x_min, pixel_size, 0, y_max, 0, -pixel_size))  # 设置地理变换参数
band = target_ds.GetRasterBand(1)
band.SetNoDataValue(0)  # 设置无效值

gdal.RasterizeLayer(target_ds, [1], lyr, burn_values=[255])  # 将几何对象栅格化到栅格数据集中

target_ds.FlushCache()