创建Shapefile文件并写入数据
作者:阿振
邮箱:tanzhenyugis@163.com
博客:https://blog.csdn.net/theonegis/article/details/80089375
修改时间:2018-06-02
声明:本文为博主原创文章,转载请注明原文出处
基本思路
使用GDAL创建Shapefile数据的基本步骤如下:
- 使用
osgeo.ogr.Driver
的CreateDataSource()
方法创建osgeo.ogr.DataSource
矢量数据集
- 使用
osgeo.ogr.DataSource
的CreateLayer()
方法创建一个图层
- 使用
osgeo.ogr.FieldDefn()
定义Shapefile文件的属性字段
- 创建
osgeo.ogr.Feature
对象,设置每个属性字段的值,使用Feature
对象的SetGeometry()
定义几何属性
- 创建
Feature
对象以后,使用osgeo.ogr.Layer
的CreateFeature()
添加Feature
对象到当前图层
- 重复步骤4和5依次添加所有的
Feature
到当前图层即可
代码实现
下面的例子中,我们读取GeoJSON表示的中国省区数据,然后其转为Shapefile格式。
GeoJSON编码片段如下:
可以看到每个Feature都有一个properties字段和geometry字段,我们需要根据properties字段的信息创建Shapefile数据的属性表,根据geometry字段创建Shapefile中的几何数据。
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46
| from osgeo import ogr from osgeo import osr import json import os os.environ['SHAPE_ENCODING'] = "utf-8"
with open('China.json') as f: china = json.load(f)
driver = ogr.GetDriverByName('ESRI Shapefile') ds = driver.CreateDataSource('China.shp')
srs = osr.SpatialReference() srs.ImportFromEPSG(4326)
layer = ds.CreateLayer('province', srs, ogr.wkbPolygon)
fname = ogr.FieldDefn('Name', ogr.OFTString) fname.SetWidth(24) layer.CreateField(fname) fcx = ogr.FieldDefn('CenterX', ogr.OFTReal) layer.CreateField(fcx) fcy = ogr.FieldDefn('CenterY', ogr.OFTReal) layer.CreateField(fcy)
for f in china['features']: feature = ogr.Feature(layer.GetLayerDefn()) feature.SetField('Name', f['properties']['name']) feature.SetField('CenterX', f['properties']['cp'][0]) feature.SetField('CenterY', f['properties']['cp'][1])
polygon = ogr.CreateGeometryFromJson(str(f['geometry'])) feature.SetGeometry(polygon) layer.CreateFeature(feature) del feature ds.FlushCache()
del ds
|