读CSV文件并写arcgis shp文件

一、在这里我用到的csv文件是包含x,y坐标及高程、降雨量数据的文件。如下图所示。

二、SF简介

简单要素模型(Simple Feature，SF)，是 OGC 国际组织定义的面向对象的矢量数据模
型。要素 Feature 包括几何对象和属性信息两部分。其中，几何对象有两种表示
方式 - WKT（Well Known Text）和 WKB（Well Known Binary），前者用于编程赋值，后者
用于数据库或二进制文件格式。

三、创建SF对象

from osgeo import ogr
#SF 对象的 WKT 表示格式
pnt_wkt = "POINT(1 1)"
lin_wkt = "LINESTRING(0 0,15 15) "
pol_wkt = "POLYGON((1 1,1 10,10 10,10 1,1 1),( 2 2,2 8,8 8,8 2,2 2)) "
#创建 SF 对象
pnt = ogr.CreateGeometryFromWkt(pnt_wkt)
lin = ogr.CreateGeometryFromWkt(lin_wkt)
pol = ogr.CreateGeometryFromWkt(pol_wkt)

四、读CSV文件并写arcgis shp文件

#读csv文件

filename='stations.csv'
fh=open(filename,'r')

#读csv文件头

title=fh.readline()

title=title.strip()

title=title.split(",")

#读取csv数据

vals=[]

datas=[]

val=fh.readline()

while val is not None:
　　val=val.strip()
　　if len(val) ==0:
　　　　break
　　for v in val.split(","):
　　　　v_=float(v)
　　　　vals.append(v_)
　　datas.append(vals)
　　vals=[]
　　val=fh.readline()

#写shp文件

from osgeo import ogr,osr

import os

filename = "stations.shp"

driver = ogr.GetDriverByName("ESRI Shapefile")
#判断shp文件名是否在当前文件夹中存在,如果存在则删除
if os.access(filename, os.F_OK ):
　　driver.DeleteDataSource(filename)

#创建shp文件

ds = driver.CreateDataSource(filename)

#定义投影

spatialref=osr.SpatialReference('LOCAL_CS["arbitrary"]')

#定义几何类型

geomtype=ogr.wkbPoint

#定义图层（此方法可以根据datas数据设定字符编码默认为utf-8编码。）

layer=ds.CreateLayer(filename[:-4],srs=spatialref,geom_type=geomtype)

#定义图层属性字段信息

fdx=ogr.FieldDefn(title[0],ogr.OFTReal)
fdx.SetPrecision(3)
fdy=ogr.FieldDefn(title[1],ogr.OFTReal)
fdy.SetPrecision(3)
fdz=ogr.FieldDefn(title[2],ogr.OFTReal)
fdz.SetPrecision(3)
fdp=ogr.FieldDefn(title[3],ogr.OFTReal)
fdp.SetPrecision(3)

layer.CreateField(fdx)

layer.CreateField(fdy)

layer.CreateField(fdz)

layer.CreateField(fdp)

#将几何对象及属性写入图层

for i in range(len(datas)):
　　d=datas[i]

　　#创建SF对象
　　wkt="POINT(%f %f %f)" % (d[0],d[1],d[2])
　　geom=ogr.CreateGeometryFromWkt(wkt)

　　#图层属性信息
　　feat=ogr.Feature(layer.GetLayerDefn())
　　feat.SetGeometry(geom)
　　feat.SetField(title[0],d[0])
　　feat.SetField(title[1],d[1])
　　feat.SetField(title[2],d[2])
　　feat.SetField(title[3],d[3])
　　layer.CreateFeature(feat)

#关闭文件

ds.Dsetroy()