1、除了一些转为shapefile文件必要的设置外,python读取csv文件也有一些需要注意的地方。
(1)首先读取了第一行fds作为字段名
(2)之后以此读取文件内容存在data里,这部分代码比较冗长
(3)代码容易出错的地方在于ds = driver.CreateDataSource(filename[:-4])这行代码,生成shapefile文件的过程,首先是创建给定名称的文件夹(我这里是stations2),然后再在文件夹里储存相应的shp等文件,文件夹的名称和子文件的名称可以不同,但是注意一定不能写成ds = driver.CreateDataSource(filename),否则以filename为名称的原csv文件会被删除。成功创建了文件夹后,就可以创建图层了,layer = ds.CreateLayer(filename [:-4], srs=spatialref, geom_type=geomtype,options=[‘ENCODING=UTF-8‘])这一句代码之前一直报错,之后又是怎么调好了不太清楚,但是和路径里有没有中文没有关系。
(4)以下代码只写入x,y和precipitation
2、
##给定一个采样点文件(格式:csv) 将其转换为shape文本
from osgeo import ogr
from osgeo import osr
import os
#############read csv file############
filename=‘stations2.csv‘
fh=open(filename,‘r‘)
lns=fh.readline()
lns=lns.strip()
fds=lns.split(‘,‘)
data=[]
lns=fh.readline()
vals=[]
while lns is not None:
lns=lns.strip()
if len(lns)==0:
break
for v in lns.split(‘,‘):
#print(v)
v_ = float(v)
vals.append(v_)
data.append(vals)
lns=fh.readline()
vals=[]
fh.close()
############create shapefile#############
driver=ogr.GetDriverByName("ESRI Shapefile")
#如文件已存在,则删除
shapename="stations2.shp"
if os.access(shapename,os.F_OK):
driver.DeleteDataSource(shapename)
#创建 Shape 文件
ds = driver.CreateDataSource(filename[:-4])
spatialref = osr.SpatialReference( ‘LOCAL_CS["arbitrary"]‘ )
geomtype=ogr.wkbPoint#点图层
layer = ds.CreateLayer(filename [:-4], srs=spatialref,
geom_type=geomtype,
options=[‘ENCODING=UTF-8‘])#创建图层,定义存数
据的编码
fd_x=ogr.FieldDefn(fds[0],ogr.OFTReal)#浮点型
fd_x.SetPrecision(3)
fd_y=ogr.FieldDefn(fds[1],ogr.OFTReal)#浮点型
fd_y.SetPrecision(3)
fd_p=ogr.FieldDefn(fds[3],ogr.OFTReal)#浮点型
fd_p.SetPrecision(3)
layer.CreateField(fd_x)
layer.CreateField(fd_y)
layer.CreateField(fd_p)
for i in range(len(data)):
d=data[i]
wkt="POINT(%f %f)"%(d[0],d[1])
geom=ogr.CreateGeometryFromWkt(wkt)
feat=ogr.Feature(layer.GetLayerDefn())#创建SF
feat.SetGeometry(geom)
feat.SetField(fds[0],d[0])
feat.SetField(fds[1],d[1])
feat.SetField(fds[3],d[3])
layer.CreateFeature(feat)
ds.Destroy()
原文地址:https://www.cnblogs.com/vividautumn/p/11613118.html