Python+OGR库学习(三):将含有字段值的TXT文件写入shpfile


代码任务

已知有一个TXT文件,每行格式:country_name:X1 Y1,X2 Y2,…Xn,Yn,其中,Xn,Yn指折点坐标,把它导出为shp文件,TXT如下:
在这里插入图片描述

代码思路

1、导入相关库包,切换到当前文件夹
2、注册驱动,创建矢量文件,获取图层
3、创建输出文件,并获取图层(没有属性定义)
4、定义输出图层字段属性:
(1)TXT文件中只能得到一个字段,命名为name,设置属性
(2)打开TXT,读取遍历每一行:
创建ring类型geom,分割每行内容为两部分(name和x,y值),将这行所有x,y值添加为ring坐标点
创建polygon,将含有坐标的ring添加进去构成一个polygon
创建这一行对应的一个要素,并写入name字段值
要素写入图层
清除此循环中ring、polygon及feature缓存,便于下次循环
(3)清除DataSource并关闭TXT文件

关键点

每行创建一个要素

polygon创建思路:ring-addpoint to ring -polygon–addring to polygon

split分割字段和坐标点

代码


#导入相关库包
import osgeo,gdal,os,ogr,osr
#创建输出shp
driver = ogr.GetDriverByName('ESRI Shapefile')
os.chdir(r'F:\PQ\python_exercise\7weeks数据\ospy_data2\ospy_data2')
if os.path.exists('country.shp'):
    driver.DeleteDataSource('country.shp')
outds = driver.CreateDataSource('country.shp')
if outds == None:
    print('创建文件失败!')
#创建图层
dst_osr = osr.SpatialReference()
dst_osr.ImportFromEPSG(4326)
outlayer = outds.CreateLayer('country',dst_osr,geom_type = ogr.wkbPolygon)
#设置字段属性
fieldDefn = ogr.FieldDefn('name',ogr.OFTString)
fieldDefn.SetWidth(50)
#图层创建属性
outlayer.CreateField(fieldDefn)
#读取要素统一属性
featuredefn = outlayer.GetLayerDefn()

#读取TXT
txtfile = open('ut_counties.txt','r')
#print(txtfile)
#遍历每一行,对每一行创建一个要素,创建polygon对象,写入对象及属性
for line in txtfile:
    #print(line)
    #创建polygon对象的ring
    ring = ogr.Geometry(ogr.wkbLinearRing)
    #获取TXT中的点坐标,用于ring的坐标点
    tmp = line.split(':')
    name = tmp[0]
    coords = tmp[1]
    coordslist = coords.split(',')#得到坐标点对
    print(coordslist)
    for coord in coordslist:
        xy = coord.split()#获取坐标点
        x = float(xy[0])
        y = float(xy[1])
        #给ring添加点坐标
        ring.AddPoint(x,y)
    #创建polygon对象
    polygon = ogr.Geometry(ogr.wkbPolygon)
    polygon.AddGeometry(ring)
    #创建要素(使用要素统一属性),写入几何图形,字段属性
    feature = ogr.Feature(featuredefn)
    feature.SetGeometry(polygon)
    feature.SetField('name',name)
    #要素写入图层
    outlayer.CreateFeature(feature)
    #每次写入完,销毁ring,polygon和feature
    ring.Destroy()
    polygon.Destroy()
    feature.Destroy()
#TXT文件关闭和DataSource清除
txtfile.close()
outds.Destroy()

结果

在这里插入图片描述

学习使用,如有意见,感谢指正*

转载自:https://blog.csdn.net/weixin_42924891/article/details/85333559

You may also like...