前言
最近使用GDAL对shape文件进行查询，走了很多弯路，查看网上的代码，发现很多代码都是错误的，或者不详细，浪费了很多时间。这里简单记录一下使用GDAL查询和读取要素的代码。
python版本3.5
GDAL版本2.1.0
代码实现
读取shape代码
from osgeo import ogr
# 注册所有的驱动
ogr.RegisterAll()
ds = ogr.Open(path, 0)
if ds is None: # 打开失败
print "打开失败"
return
# 获取该数据源中的图层个数，一般shp数据图层只有一个，如果是mdb、dxf等图层就
在GIS（地理信息系统）领域，处理空间数据时经常会用到各种格式的文件，其中Shapefile是一种常见的矢量数据格式。本篇文章将详细讲解如何利用Python编程语言和GDAL（Geospatial Data Abstraction Library）库来对Shapefile进行查询和读取要素。
GDAL是一个强大的开源库，用于处理多种地理空间数据格式，包括Shapefile。在Python中使用GDAL，我们需要导入`osgeo.ogr`模块。以下是一段基础的代码示例，展示了如何打开并检查一个Shapefile：
```python
from osgeo import ogr
# 注册所有驱动器
ogr.RegisterAll()
# 打开Shapefile，第二个参数0表示只读模式
ds = ogr.Open(path, 0)
if ds is None:
print("打开失败")
return
# 获取数据源中的图层数量
layer_count = ds.GetLayerCount()
if layer_count != 1:
print("图层数量异常")
```
在上述代码中，`ogr.Open()`函数用于打开Shapefile，返回一个数据源对象（`ds`）。如果文件无法打开，`ds`将为`None`。接着，我们检查图层数量，通常Shapefile只有一个图层，但其他数据源可能包含多个。
接下来，我们可以对Shapefile进行查询。以下是一个例子，展示如何筛选出`zwdm`字段等于“07”的记录：
```python
# 获取第一个图层
layer = ds.GetLayerByIndex(0)
# 重置图层读取，清除之前的过滤
layer.ResetReading()
# 设置属性过滤器
status = layer.SetAttributeFilter("zwdm = '07'")
if status == 0:
print("查询成功")
else:
print("查询失败")
# 获取匹配记录的数量
feature_count = layer.GetFeatureCount()
print("匹配记录数:", feature_count)
# 逐个读取匹配的要素
feature = layer.GetNextFeature()
while feature is not None:
# 获取要素属性
zwdm = feature.GetFieldAsString("zwdm")
# 执行你的操作...
# 读取下一个要素
feature = layer.GetNextFeature()
```
在上述代码中，`SetAttributeFilter()`函数用于设置查询条件，`GetFeatureCount()`获取匹配记录的总数。需要注意的是，当设置了过滤器后，`GetNextFeature()`只会返回满足条件的要素。如果在循环中忘记调用`GetNextFeature()`，会导致无限循环。
在实际操作中，`GetFieldAsString()`用于获取字段值，这里的`zwdm`是字段名。你可以根据实际需求替换这个字段名，进行相应的操作，例如数据处理、可视化或其他分析。
通过Python和GDAL，我们可以方便地对Shapefile进行查询和读取，这对于GIS开发和数据分析非常有用。在进行这些操作时，确保正确处理数据源、图层、过滤器和特征，避免遇到不必要的问题。同时，理解GDAL库的其他功能，如几何操作、投影转换等，能帮助我们更全面地处理空间数据。
