shp文件按属性删除

 

from osgeo import ogr, osr, gdal ,gdalconst
import numpy as np
import math
import sympy as sy
import cv2

#注册所有驱动
gdal.AllRegister()
#解决中文路径乱码问题
gdal.SetConfigOption("GDAL_FILENAME_IS_UTF8", "NO")

def Attribute_filtering(shp,condition):
    driver = ogr.GetDriverByName('ESRI Shapefile')
    pFeatureDataset = driver.Open(shp, 1)
    pFeaturelayer = pFeatureDataset.GetLayer(0)
    # 输入属性条件
    pFeaturelayer.SetAttributeFilter(condition)
    pFeatureDef = pFeaturelayer.GetLayerDefn()
    pLayerName = pFeaturelayer.GetName()
    pFieldName = "Value"
    pFieldIndex = pFeatureDef.GetFieldIndex(pFieldName)
    for pFeature in pFeaturelayer:
        pFeatureFID = pFeature.GetFID()
        pFeaturelayer.DeleteFeature(int(pFeatureFID))



if __name__ == '__main__':
    shp=r"E:\jpg_test\test2\output\test.shp"
    #属性过滤
    condition="Value >0"
    Attribute_filtering(shp,condition)