基于python和GDAL实现对shape文件要素进行查询

前言

最近使用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等图层就会有多个
 ilayercount = ds.GetLayerCount()
if ilayercount != 1:
    print("图层数量异常")

注意，网上有人写的为了支持中文路径，代码里要加下面的代码，我测试中文路径是不需要添加下面的代码的

        # # 为了支持中文路径，请添加下面这句代码
        # gdal.SetConfigOption("GDAL_FILENAME_IS_UTF8", "NO")
        # # 为了使属性表字段支持中文，请添加下面这句
        # gdal.SetConfigOption("SHAPE_ENCODING", "")

对shape做查询代码

下面以对shape中zwdm字段等于“07”的记录做筛选你

 # 获取第一个图层，一般shape只有一层
olayer = ds.GetLayerByIndex(0)
 # 对图层进行初始化，如果对图层进行了过滤操作，执行这句后，之前的过滤全部清空
olayer.ResetReading()
# 进行查询，如果status=0，则说明查询生效，
status = olayer.SetAttributeFilter("zwdm= '07'")

注意，即使改shape有500条记录，只要筛选成功，就默认shape中只有符合条件的记录，逐个读取这个shape也无法读取到其他的记录。

获取要素数量

olayer.GetFeatureCount()

网上有人写后面括号里要输入0，亲测输入0时，获取的数量永远是-1。
对图层进行查询后，此处获取的是查询后的数量

逐个读取要素

        ofeature = olayer.GetNextFeature() # 读取第一个要素
        # 下面开始遍历图层中的要素
        while ofeature is not None:
            # 获取要素中的属性表内容
            zwdm = ofeature.GetFieldAsString("zwdm")
            # ----各类操作

            # 读取下一个要素
            ofeature = olayer.GetNextFeature()

这个读取的代码是比较坑的，之前看网上的代码，while里没有写读取下一个要素的操作，我还以为是自动会读取下一个要素，结果写出来的代码无限循环了。

结束

最后希望大家少采坑吧