使用pyshp将.csv文件转换为.shp？

10

我试图了解如何在python中使用csv模块在与python脚本相同的文件夹中打开csv文件，然后使用shapefile模块pyshp创建shapefile。

csv文件如下所示，但可以有几千行的记录：

id_nr;date;target;start_lat;start_lon
1;2012-05-21;navpoint 25x;55.123654;13.456954
1;2012-05-23;navpoint 11f;55.143654;12.456954

— 科贾
source

14

pyshp模块很难掌握，但是一旦开始使用，它确实很有用。我编写了一个脚本，该脚本读取示例数据的csv并写出一个shapefile，并将数据存储为正确数据类型的属性。在我找到本本关于xbase格式的用户指南之前，pyshp / xbase数据类型一直对我来说很棘手，由于这个问题，我在自己的博客上写了一条有关pyshp数据类型的小注释，下面将其中部分内容进行了粘贴。：

C是ASCII字符
N是一个双精度整数，长度限制为大约18个字符
D是YYYYMMDD格式的日期，各节之间没有空格或连字符。
F是与N具有相同长度限制的浮点数
L是用于逻辑数据的逻辑数据，该逻辑数据作为1（true）或0（false）的短整数存储在shapefile的属性表中。它可以接收的值为1、0，y，n，Y，N，T，F或python内置的True和False

完整列表如下：

import shapefile as shp
import csv

out_file = 'GPS_Pts.shp'

#Set up blank lists for data
x,y,id_no,date,target=[],[],[],[],[]

#read data from csv file and store in lists
with open('input.csv', 'rb') as csvfile:
    r = csv.reader(csvfile, delimiter=';')
    for i,row in enumerate(r):
        if i > 0: #skip header
            x.append(float(row[3]))
            y.append(float(row[4]))
            id_no.append(row[0])
            date.append(''.join(row[1].split('-')))#formats the date correctly
            target.append(row[2])

#Set up shapefile writer and create empty fields
w = shp.Writer(shp.POINT)
w.autoBalance = 1 #ensures gemoetry and attributes match
w.field('X','F',10,8)
w.field('Y','F',10,8)
w.field('Date','D')
w.field('Target','C',50)
w.field('ID','N')

#loop through the data and write the shapefile
for j,k in enumerate(x):
    w.point(k,y[j]) #write the geometry
    w.record(k,y[j],date[j], target[j], id_no[j]) #write the attributes

#Save shapefile
w.save(out_file)

我希望这有帮助。

— 伤心
source

非常好的脚本。我收到一个错误，因为它没有将其作为文本读取，所以我更改了这一行：使用open（'input.csv'，'rt'）作为csvfile：

— 逆流

1

我认为您可以通过在for循环之前使用next（r）跳过标题而不是使用if语句进行检查来提高性能。

— 罗维科

@sgrieve-此脚本转换具有特定预定字段的csv。我想要将任何csv转换为要素类的通用脚本。也许有一些有用的arcpy函数可以实现这一目标？

— 沃特曼

2

或者，您不需要将数据保存在列表中。

# import libraries
import shapefile, csv

# create a point shapefile
output_shp = shapefile.Writer(shapefile.POINT)
# for every record there must be a corresponding geometry.
output_shp.autoBalance = 1
# create the field names and data type for each.
# you can insert or omit lat-long here
output_shp('Date','D')
output_shp('Target','C',50)
output_shp('ID','N')
# count the features
counter = 1
# access the CSV file
with open('input.csv', 'rb') as csvfile:
    reader = csv.reader(csvfile, delimiter=',')
    # skip the header
    next(reader, None)
    #loop through each of the rows and assign the attributes to variables
    for row in reader:
        id= row[0]
        target= row[1]
        date = row[2]
        # create the point geometry
        output_shp.point(float(longitude),float(latitude))
        # add attribute data
        output_shp.record(id, target, date)
        print "Feature " + str(counter) + " added to Shapefile."
        counter = counter + 1
# save the Shapefile
output_shp.save("output.shp")

您可以在此处找到此实现的有效示例。

— Clubdebambos
source