在使用 Python 将矢量面数据转换为栅格数据时，通常会利用 `GDAL` 和 `OGR` 库来处理地理空间数据。矢量数据（如 Shapefile）包含几何对象（点、线、面），而栅格数据则是像素图像，每个像素具有一个特定的值。转换过程包括定义输出栅格的大小、投影、分辨率，以及将矢量面的属性值烧录到栅格中。 以下是实现矢量面数据转换为栅格数据的步骤和示例代码： ### 1. 准备工作 - **矢量文件**：例如 `input.shp`，包含面要素。 - **输出栅格文件**：指定输出文件路径，如 `output.tif`。 - **目标分辨率**：设定输出栅格的像元大小，例如 `pixel_size = 30`。 - **目标投影**：可指定输出栅格的投影系统，如 `EPSG:4326`。 ### 2. Python 实现代码 ```python from osgeo import gdal, ogr, osr def vector_to_raster(vector_path, raster_path, attribute_name, pixel_size=30): # 打开矢量数据 driver = ogr.GetDriverByName('ESRI Shapefile') vector_dataset = driver.Open(vector_path, 0) layer = vector_dataset.GetLayer() # 获取矢量数据的空间范围 x_min, x_max, y_min, y_max = layer.GetExtent() # 计算栅格尺寸 cols = int((x_max - x_min) / pixel_size) + 1 rows = int((y_max - y_min) / pixel_size) + 1 # 创建栅格数据集 raster_driver = gdal.GetDriverByName('GTiff') raster_dataset = raster_driver.Create(raster_path, cols, rows, 1, gdal.GDT_Byte) raster_dataset.SetGeoTransform((x_min, pixel_size, 0, y_max, 0, -pixel_size)) # 设置投影 srs = layer.GetSpatialRef() if srs is None: srs = osr.SpatialReference() srs.ImportFromEPSG(4326) # 默认使用 WGS84 raster_dataset.SetProjection(srs.ExportToWkt()) # 初始化栅格值为 NoData band = raster_dataset.GetRasterBand(1) band.SetNoDataValue(0) # 使用 GDAL 的 RasterizeLayer 方法进行矢量转栅格 err = gdal.RasterizeLayer(raster_dataset, [1], layer, options=[f'ATTRIBUTE={attribute_name}']) if err != 0: raise ValueError("栅格化失败") # 清理资源 raster_dataset = None vector_dataset = None # 示例调用 vector_to_raster('input.shp', 'output.tif', 'class', pixel_size=30) ``` ### 3. 参数说明 - `vector_path`：矢量文件路径（如 Shapefile）。 - `raster_path`：输出栅格文件路径（如 GeoTIFF 格式）。 - `attribute_name`：用于栅格值的字段名称，例如 `'class'`。 - `pixel_size`：栅格像元大小，默认为 30 米。 ### 4. 注意事项 - 确保矢量数据和输出栅格的投影一致，否则需要进行坐标转换。 - 若矢量数据中存在多个面要素，`RasterizeLayer` 会根据 `attribute_name` 的值分别烧录到栅格中。 - 可以调整 `pixel_size` 来控制输出栅格的分辨率，较小的值会增加文件大小和计算量。 ### 5. 依赖库安装 - 需要安装 `GDAL` 和 `OGR`，可以通过以下命令安装： ```bash pip install GDAL ``` ---