python ogr_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