码迷,mamicode.com
首页 > 其他好文 > 详细

命令行记录-csv转为shape文件

时间:2019-09-30 16:53:18      阅读:99      评论:0      收藏:0      [点我收藏+]

标签:geometry   station   line   浮点型   float   spl   driver   val   ad cs   

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()

命令行记录-csv转为shape文件

标签:geometry   station   line   浮点型   float   spl   driver   val   ad cs   

原文地址:https://www.cnblogs.com/vividautumn/p/11613118.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!