Python与开源GIS：根据属性条件选择要素

根据属性来对数据库中的记录进行选择是数据库应用中必不 可少的一项功能，对于地理数据，也是如此，譬如人口在百 万以上的城市，面积在百万平方千米以上的国家。

根据属性条件选择要素

Layer对象有一个方法叫SetAttributeFilter()，可以将根据 属性，将Layer中符合某一条件的Feature过滤出来。设定了 Filter之后就可以用GetNextFeature()方法依次取出符合条 件的Feature了。

    from osgeo import ogr
    import os
    shpfile = '/gdata/world_borders.shp'
    ds = ogr.Open(shpfile)
    layer = ds.GetLayer(0)
    lyr_count = layer.GetFeatureCount()
    print(lyr_count)
    layer.SetAttributeFilter("AREA > 800000")
    lyr_count = layer.GetFeatureCount()
    print(lyr_count)

上面的代码首先将图层中的要素的数目打印出来。 然后，使用过滤器，来选择面积在80万平方公里以上的国家。这里使用了“AREA”字段，这个字段是早已经在Shapefile 中生成好的。 然后再打印图层中要素的数目，就会发现 数据变少了。

根据属性条件生成要素

大多数时候，选择要素的目的并不是为了简单地查看一下 数目， 而是要将数据保存了，以方便以后的使用。请仔细查看下面的代码。注意，生成矢量数据可以在不同的 层次上。

    driver = ogr.GetDriverByName("ESRI Shapefile")
    extfile = '/v/filter_demo.shp'
    if os.access( extfile, os.F_OK ):
    driver.DeleteDataSource( extfile )
    newds = driver.CreateDataSource(extfile)
    layernew = newds.CreateLayer('rect',None,ogr.wkbPolygon)
    feat = layer.GetNextFeature()
    while feat is not None:
    layernew.CreateFeature(feat)
    feat = layer.GetNextFeature()
    newds.Destroy()

看一下生成的结果吧。这是一幅不完整的世界地图，因为 它只包含了面积在80万平方公里以上的国家。

对于这个结果，首先要注意的就是有一些面积比较小的图斑 也被选了出来，这些图斑是属于某些国家的，其面积定义是 与某个大图斑一致的，也就是这此处的面积是国家的面积，而并非是图斑的面积；另外要注意的是台湾并没有被选择 出来：这是个政治问题，而非技术问题。

返回目录：Python与开源GIS