python矢量shp转栅格

在Python中，要将Shapefile（.shp）这种常见的地理空间数据格式转换为栅格（Raster），你可以使用GDAL库。GDAL提供了广泛的地理空间处理功能，包括数据读写、转换等。

首先，确保你已经安装了GDAL和 Fiona，这两个库分别用于高级地理空间操作和与Shapefile交互：

pip install gdal Fiona

然后，你可以编写一个Python脚本来执行这个转换过程。以下是一个简单的示例，假设我们有一个名为`input.shp`的Shapefile，你想将其转换为栅格并保存到一个新的tif文件中，栅格分辨率设为10米：

from osgeo import ogr, gdal

# 设置源Shapefile路径和输出栅格文件路径
source_file = "path_to_your_input.shp"
output_raster = "path_to_save_output_raster.tif"

# 创建数据驱动器对象
driver = ogr.GetDriverByName("ESRI Shapefile")

# 打开Shapefile
dataSource = driver.Open(source_file, 0)

# 获取第一个层（通常只有一个）
layer = dataSource.GetLayer()

# 获取几何字段类型（例如，如果Shapefile包含点，则是"POINT"）
geometry_type = layer.GetGeomType()

# 创建栅格驱动（这里假设我们要用GTiff）
raster_driver = gdal.GetDriverByName("GTiff")

# 创建输出数据集（带GeoTransform和Projection）
width = layer.XCount
height = layer.YCount
bands = 1
projection = layer.GetSpatialRef()
geotransform = layer.GetGeometryRef().GetEnvelope()
output_dataset = raster_driver.Create(output_raster, width, height, bands, gdal.GDT_Float32, options=["COMPRESS=LZW"])

# 写入元数据
output_dataset.SetProjection(projection.ExportToWkt())
output_dataset.SetGeoTransform(geotransform)

# 遍历每个Shapefile特征并创建栅格值
for feature in layer:
    geom = feature.GetGeometryRef()
    x, y = geom.GetX(), geom.GetY()  # 根据几何类型调整坐标获取方式
    output_dataset.WriteArray([x, y], x, y)  # 如果是点，可能直接是坐标；如果是线或面，可能需要采样或计算中心点

# 保存并关闭数据集
output_dataset.FlushCache()
output_dataset = None

# 关闭Shapefile
dataSource = None

print("Shapefile converted to raster successfully.")

运行这段代码之前，请确认路径和参数正确无误。如果你的Shapefile中的几何类型不是预期的点，可能需要对获取栅格值的部分进行相应调整。