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

Posted 2022-12-06 vividautumn

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