Python：Shapefile矢量转化为GeoJSON格式

在最近的项目中，完成了许多python处理矢量数据的算法程序，比如缓冲区分析、叠置分析、统计分析等，主要用到的是GDAL/OGR库，很多功能都参照了此链接中的示例：http://pcjericks.github.io/py-gdalogr-cookbook/#

由于完成的功能在上面链接中都能找到实现的代码，所以原来不打算写博客的，但是今天写的这个SHP转GeoJSON的算法把我坑得不轻，着实郁闷了好久才解决，因此就写一下权当纪念吧。有时候写程序真的是一件充满玄学的事情，原理上怎么看怎么对，但是运行起来却偏偏不是那回事，就算调试也给你不报错直接闪退，特别是处理空间数据时属性的编码问题，更是搞得人头皮发麻。这种情况下，能不能调的出来全靠运气，搞不好一个小问题卡一个星期都是正常，这可能也是程序员普遍发量稀少的原因吧。更加玄学的是，解决问题的方式又常常充满戏剧性，折腾了一个早上加半个下午没找到原因，起来喝杯水却突然灵感乍现，三四行代码调整一下顺序，一顿骚操作，点击运行，突然就能运行成功了，再看一下结果，还真是对的！山穷水尽疑无路，柳暗花明又一村，这种感觉真是刺激。

好了，将Shapefile转化为GeoJSON的完整代码如下：

1 1 # -*- coding: utf-8 -*-
2 2 from osgeo import ogr
3 3 import gdal
4 4 import sys
5 5 import os
6 6 
7 7 def ChangeToJson(vector, output):
8 8     print("Starting........")
9 9     #打开矢量图层
1010     gdal.SetConfigOption("GDAL_FILENAME_IS_UTF8", "YES")
1111     gdal.SetConfigOption("SHAPE_ENCODING", "GBK")
1212     shp_ds = ogr.Open(vector)
1313     shp_lyr = shp_ds.GetLayer(0)
1414 
1515     # 创建结果Geojson
1616     baseName = os.path.basename(output)
1717     out_driver = ogr.GetDriverByName('GeoJSON')
1818     out_ds = out_driver.CreateDataSource(output)
1919     if out_ds.GetLayer(baseName):
2020         out_ds.DeleteLayer(baseName)
2121     out_lyr = out_ds.CreateLayer(baseName, shp_lyr.GetSpatialRef())
2222     out_lyr.CreateFields(shp_lyr.schema)
2323     out_feat = ogr.Feature(out_lyr.GetLayerDefn())
2424 
2525     #生成结果文件
2626     for feature in shp_lyr:
2727         out_feat.SetGeometry(feature.geometry())
2828         for j in range(feature.GetFieldCount()):
2929             out_feat.SetField(j, feature.GetField(j))
3030         out_lyr.CreateFeature(out_feat)
3131 
3232     del out_ds
3333     del shp_ds
3434     print("Success........")
3535 
3636 if __name__ == '__main__':
3737     shapefile = 'D:/Python37Work/HainanAOI/affected_bridge.shp'
3838     out = 'D:/Python37Work/HainanAOI/affected_bridge.json'
3939     ChangeToJson(shapefile, out)

有以下几个注意事项：

1、gdal.SetConfigOption("GDAL_FILENAME_IS_UTF8", "YES")和gdal.SetConfigOption("SHAPE_ENCODING", "GBK")

这两行用于支持中文，前一句用于支持文件名称及路径内的中文，后一句用于支持属性字段中的中文，最好将其放在最前面，否则读取矢量文件时不能按GBK编码方式读取。

2、out_lyr.CreateFields(shp_lyr.schema)和out_feat = ogr.Feature(out_lyr.GetLayerDefn())

这两句的顺序一定不能错，否则会导致部分字段无法写入，应该是先对out_lyr创建相应的字段，然后再由out_feat来一个个生成Feature。

最终生成的.json文件内容如下：

Python：Shapefile矢量转化为GeoJSON格式

评论区

相关推荐