GDAL 实现矢量数据读写

❝ GIS 数据的读写作为一个基础操作，是每一个GISer的必修课。在使用GDAL读取矢量数据时，需要掌握其基本的数据结构与类型，了解常用的数据读取方法，这样开发时才会起到事半功倍的效果。

在之前的文章中讲了如何使用GDAL或者ogr2ogr工具将txt以及csv文本数据转换为Shp格式，本篇教程在之前一系列文章的基础上讲解GDAL实现矢量数据读写。

如果你还没有看过，建议从以上内容开始。

1. 开发环境

本文使用如下开发环境，以供参考。

时间：2025年

系统：Windows 11

Python：3.11.7

GDAL：3.11.1

2. 矢量数据读取接口

在GDAL中，矢量数据模型遵循下图所示结构，数据驱动->数据源->图层->属性。

2.1. 获取数据源

GDAL中可以使用Open方法直接打开数据源，进而读取图层数据。在GDAL Python API中，Driver和Dataset 类在Raster API文档中，既适用于矢量数据，也适用于栅格数据。

python 复制代码

"""
说明：用于打开数据源
参数：
    -utf8_path：字符串，数据源路径
    -update：布尔值类型，默认值False，是否以更新模式打开数据集（默认为只读模式）
返回值：返回数据集或者None
"""
def Open(utf8_path,update=False)

如下以只读模式打开Shp数据源。

ini 复制代码

dataSource = ogr.Open(shpPath,False) # False表示只读，True表示可写

除了ogr的Open方法之外，也可以使用gdal模块的打开方法

2.2. 获取图层

GDAL获取图层的常用方法为GetLayer，该方法只有一个参数。iLayer值固定为0，对shapefile图层来说是可选项。

python 复制代码

"""
说明：从数据集获取图层
参数：
    -iLayer：整型或字符串，图层名称或者基于0的图层索引值。
返回值：返回图层或者None
"""
def GetLayer(iLayer=0)

获取数据集中索引值为0的图层。

ini 复制代码

 # 获取图层
layer = dataSource.GetLayer(0)

除了GetLayer方法之外，还有一个GetLayerByIndex方法用于获取图层。

python 复制代码

"""
说明：根据索引获取图层
参数：
    -index：整型，图层索引值。
返回值：返回图层或者None
"""
def GetLayerByIndex(index)

图层索引值处于0和GetLayerCount() - 1之间。

最后还有个GetLayerCount方法用于获取图层数量。

从数据集中获取图层数量。

python 复制代码

 # 获取图层数量
layerCount = dataset.GetLayerCount()
print(f"图层数量：{layerCount}")

2.3. 获取投影信息

GetProjection和GetProjectionRef方法都可以用于读取坐标系统，两者都返回一个字符串类型WKT格式的数据集空间参考信息。

从图层获取投影信息。

python 复制代码

 # 图层投影
projectionRef = layer.GetProjectionRef()
print(f"空间参考：{projectionRef}")

此外还有一个方法GetSpatialRef也可用于获取数据集空间参考信息。

从图层获取空间参考信息。

python 复制代码

 # 图层空间参考
spatialRef = layer.GetSpatialRef()
print(f"空间参考：{spatialRef}")

2.4. 获取要素

在知道FId(要素编号)的情况下，可以直接使用GetFeature方法获取要素。

ini 复制代码

 # 获取指定要素
feature = layer.GetFeature(0)

也可以通过GetNextFeature方法遍历要素特征，并且返回一个要素Feature。此方法适用于少数几个OGRLayer.GetNextFeature()效率不高的驱动程序，但总体而言，OGRLayer.GetNextFeature()是一个更自然的API。

从图层读取要素特征。

ini 复制代码

 # 获取要素
feature = layer.GetNextFeature()
limitCount = 0
 # 限制打印前十个要素
while feature and limitCount < 10:
    print(f"打印第【{limitCount+1}】个要素")

GetFeatureCount方法用于获取图层要素数量。

python 复制代码

 # 获取图层要素数量
layerFeatureCount = layer.GetFeatureCount()
print(f"要素数量：{layerFeatureCount}")

2.5. 获取属性结构

GetLayerDefn方法用于获取图层属性表，返回要素定义信息。

获取图层属性。

ini 复制代码

 # 获取图层属性
layerProperty = layer.GetLayerDefn()

GetFieldCount方法用于获取字段数量。

读取图层要素数量。

ini 复制代码

 # 获取图层名称
layerName = layer.GetName()
 # 获取图层要素数量
layerFeatureCount = layer.GetFeatureCount()

GetFieldDefn用于获取字段定义信息，其中包括字段名称，字段类型。

读取属性表结构。

ini 复制代码

 # 获取图层属性
layerProperty = layer.GetLayerDefn()
 # 获取图层字段数量
fieldCount = layerProperty.GetFieldCount()
print(f"字段数量：{fieldCount}")

 # 获取字段信息
for j in range(fieldCount):
    # 获取字段属性对象
    fieldProperty = layerProperty.GetFieldDefn(j)
    # 获取字段属性名称
    fieldName = fieldProperty.GetName()
    # 获取字段属性类型
    fieldType = fieldProperty.GetTypeName()

    print(f"第 【{j}】 个字段名称：{fieldName}，字段类型：{fieldType}")

GetField用于获取属性字段值，参数可以是字符串字段名，也可以是字段索引值。

GetGeometryRef方法用于获取要素几何对象信息。

读取几何对象。

ini 复制代码

 # 读取几何属性
geom = feature.GetGeometryRef()

2.6. 获取图层范围

GetExtent获取图层四至范围。

ini 复制代码

 # 获取范围
extent = layer.GetExtent()

3. 矢量数据修改接口

Driver (驱动程序)是一个知道如何执行特定功能的对象与某种数据类型（如shapefile），为了读写数据，就需要一个合适的驱动程序。

本文以下的数据修改都已shapefile驱动为例进行讲解。

3.1. 获取目标数据驱动

首先需要导入ogr模块用于处理矢量数据。GetDriverByName方法可以根据名称获取数据驱动。

ini 复制代码

from osgeo import ogr,osr

 # 获取Shp数据驱动
shpDriver = ogr.GetDriverByName('ESRI Shapefile')

3.2. 创建数据源

CreateDataSource方法根据文件路径创建数据源。

ini 复制代码

 # 创建Shp数据源
shpDataSource = shpDriver.CreateDataSource(shpPath)

3.3. 创建图层

通过CreateLayer方法创建图层，该方法具有四个参数，图层名称、坐标系统以及几何类型是必传。

ini 复制代码

 # 创建点图层
layer = shpDataSource.CreateLayer("points",srs,ogr.wkbPoint)

3.4. 创建属性字段

通过CreateField方法创建属性字段。

ini 复制代码

if field not in [lonField, latField]:
    # 创建字段定义
    fieldDefn = ogr.FieldDefn(field, ogr.OFTString)
    fieldDefn.SetWidth(254)
    # 直接创建字段，不要存储 FieldDefn 对象
    layer.CreateField(fieldDefn)

3.5. 创建几何对象

要素对象方法SetGeometry可以创建几何对象。

arduino 复制代码

 # 创建几何对象
point = ogr.Geometry(ogr.wkbPoint)
point.AddPoint(lon,lat)
feature.SetGeometry(point)

3.6. 创建要素

通过图层方法CreateFeature创建要素。

ini 复制代码

 # 创建要素
feature = ogr.Feature(layer.GetLayerDefn())

 # 设置属性
for field in fieldnames:
    if field not in [lonField, latField]:
        feature.SetField(field, str(row[field]))
 # 创建几何
point = ogr.Geometry(ogr.wkbPoint)
point.AddPoint(lon,lat)
feature.SetGeometry(point)

 # 保存要素
layer.CreateFeature(feature)

4. 关闭数据源

为了防止内存泄露问题发生，在确定数据读取完成后需要关闭数据源。

ini 复制代码

 # 关闭数据源
dataSource = None

 # 关闭要素
feature.Destroy()
 # feature = None

小提示

GDAL中的方法采用大驼峰命名法

❝ OpenLayers示例数据下载，请在公众号后台回复：ol数据

全国信息化工程师－GIS 应用水平考试资料，请在公众号后台回复：GIS考试

❝GIS之路 公众号已经接入了智能助手，可以在对话框进行提问，也可以直接搜索历史文章进行查看。

GDAL 实现 GIS 数据读取转换（全）

自然资源部：我国地理信息产业总产值将超9000亿元

GIS 数据转换：使用 GDAL 将 Shp 转换为 GeoJSON 数据

GIS 数据转换：使用 GDAL 将 GeoJSON 转换为 Shp 数据

GIS 数据转换：使用 GDAL 将 TXT 转换为 Shp 数据