用python写一段矢量转栅格的代码

# 以下是一段用python写的将矢量转栅格的代码：from osgeo import gdal, ogr# 定义输入和输出矢量图层路径
in_vector = "input.shp"
out_raster = "output.tif"# 打开输入矢量图层
in_ds = ogr.Open(in_vector)
in_layer = in_ds.GetLayer()# 创建栅格数据集
out_driver = gdal.GetDriverByName("GTiff")
out_ds = out_driver.Create(out_raster, xsize, ysize, 1, gdal.GDT_UInt16)# 设置投影
out_ds.SetProjection(in_layer.GetSpatialRef().ExportToWkt)# 设置坐标系
out_ds.SetGeoTransform(in_layer.GetSpatialRef().GetGeoTransform())# rasterize
gdal.RasterizeLayer(out_ds, [1], in_layer, burn_values=[1])# 关闭数据集
out_ds = None
in_ds = None