python使用gdal对shp读取,新建和更新的实例


Posted in Python onMarch 10, 2020

昨天要处理一个shp文件,读取里面的信息,做个计算然后写到后面新建的field里面。先写个外面网上都能找到的新建和读取吧。

1.读取shp文件

#-*- coding: cp936 -*-
try:
   from osgeo import gdal
   from osgeo import ogr
exceptImportError:
   import gdal
   import ogr
 
defReadVectorFile():
   # 为了支持中文路径,请添加下面这句代码
   gdal.SetConfigOption("GDAL_FILENAME_IS_UTF8","NO")
   # 为了使属性表字段支持中文,请添加下面这句
   gdal.SetConfigOption("SHAPE_ENCODING","")
 
   strVectorFile ="E:\\Datum\\GDALCsTest\\Debug\\beijing.shp"
 
   # 注册所有的驱动
   ogr.RegisterAll()
 
   #打开数据
   ds = ogr.Open(strVectorFile, 0)
   if ds == None:
     print("打开文件【%s】失败!", strVectorFile)
     return
 
   print("打开文件【%s】成功!", strVectorFile)
 
   # 获取该数据源中的图层个数,一般shp数据图层只有一个,如果是mdb、dxf等图层就会有多个
   iLayerCount = ds.GetLayerCount()
 
   # 获取第一个图层
   oLayer = ds.GetLayerByIndex(0)
   if oLayer == None:
     print("获取第%d个图层失败!\n", 0)
     return
 
   # 对图层进行初始化,如果对图层进行了过滤操作,执行这句后,之前的过滤全部清空
   oLayer.ResetReading()
 
   # 通过属性表的SQL语句对图层中的要素进行筛选,这部分详细参考SQL查询章节内容
   oLayer.SetAttributeFilter("\"NAME99\"LIKE \"北京市市辖区\"")
 
   # 通过指定的几何对象对图层中的要素进行筛选
   #oLayer.SetSpatialFilter()
 
   # 通过指定的四至范围对图层中的要素进行筛选
   #oLayer.SetSpatialFilterRect()
 
   # 获取图层中的属性表表头并输出
   print("属性表结构信息:")
   oDefn = oLayer.GetLayerDefn()
   iFieldCount = oDefn.GetFieldCount()
   for iAttr in range(iFieldCount):
     oField =oDefn.GetFieldDefn(iAttr)
     print( "%s: %s(%d.%d)" % ( \
          oField.GetNameRef(),\
          oField.GetFieldTypeName(oField.GetType() ), \
          oField.GetWidth(),\
          oField.GetPrecision()))
 
   # 输出图层中的要素个数
   print("要素个数 = %d", oLayer.GetFeatureCount(0))
 
   oFeature = oLayer.GetNextFeature()
   # 下面开始遍历图层中的要素
   while oFeature is not None:
     print("当前处理第%d个: \n属性值:", oFeature.GetFID())
     # 获取要素中的属性表内容
     for iField inrange(iFieldCount):
       oFieldDefn =oDefn.GetFieldDefn(iField)
       line = " %s (%s) = " % ( \
            oFieldDefn.GetNameRef(),\
            ogr.GetFieldTypeName(oFieldDefn.GetType()))
 
       ifoFeature.IsFieldSet( iField ):
          line = line+ "%s" % (oFeature.GetFieldAsString( iField ) )
       else:
          line = line+ "(null)"
 
       print(line)
  
     # 获取要素中的几何体
     oGeometry =oFeature.GetGeometryRef()
 
     # 为了演示,只输出一个要素信息
     break
 
   print("数据集关闭!")

2.新建shp文件

#-*- coding: cp936 -*-
try:
   from osgeo import gdal
   from osgeo import ogr
exceptImportError:
   import gdal
   import ogr
 
defWriteVectorFile():
   # 为了支持中文路径,请添加下面这句代码
   gdal.SetConfigOption("GDAL_FILENAME_IS_UTF8","NO")
   # 为了使属性表字段支持中文,请添加下面这句
   gdal.SetConfigOption("SHAPE_ENCODING","")
 
   strVectorFile ="E:\\TestPolygon.shp"
 
   # 注册所有的驱动
   ogr.RegisterAll()
 
   # 创建数据,这里以创建ESRI的shp文件为例
   strDriverName = "ESRIShapefile"
   oDriver =ogr.GetDriverByName(strDriverName)
   if oDriver == None:
     print("%s 驱动不可用!\n", strDriverName)
     return
  
   # 创建数据源
   oDS =oDriver.CreateDataSource(strVectorFile)
   if oDS == None:
     print("创建文件【%s】失败!", strVectorFile)
     return
 
   # 创建图层,创建一个多边形图层,这里没有指定空间参考,如果需要的话,需要在这里进行指定
   papszLCO = []
   oLayer =oDS.CreateLayer("TestPolygon", None, ogr.wkbPolygon, papszLCO)
   if oLayer == None:
     print("图层创建失败!\n")
     return
 
   # 下面创建属性表
   # 先创建一个叫FieldID的整型属性
   oFieldID =ogr.FieldDefn("FieldID", ogr.OFTInteger)
   oLayer.CreateField(oFieldID, 1)
 
   # 再创建一个叫FeatureName的字符型属性,字符长度为50
   oFieldName =ogr.FieldDefn("FieldName", ogr.OFTString)
   oFieldName.SetWidth(100)
   oLayer.CreateField(oFieldName, 1)
 
   oDefn = oLayer.GetLayerDefn()
 
   # 创建三角形要素
   oFeatureTriangle = ogr.Feature(oDefn)
   oFeatureTriangle.SetField(0, 0)
   oFeatureTriangle.SetField(1, "三角形")
   geomTriangle =ogr.CreateGeometryFromWkt("POLYGON ((0 0,20 0,10 15,0 0))")
   oFeatureTriangle.SetGeometry(geomTriangle)
   oLayer.CreateFeature(oFeatureTriangle)
 
   # 创建矩形要素
   oFeatureRectangle = ogr.Feature(oDefn)
   oFeatureRectangle.SetField(0, 1)
   oFeatureRectangle.SetField(1, "矩形")
   geomRectangle =ogr.CreateGeometryFromWkt("POLYGON ((30 0,60 0,60 30,30 30,30 0))")
   oFeatureRectangle.SetGeometry(geomRectangle)
   oLayer.CreateFeature(oFeatureRectangle)
 
   # 创建五角形要素
   oFeaturePentagon = ogr.Feature(oDefn)
   oFeaturePentagon.SetField(0, 2)
   oFeaturePentagon.SetField(1, "五角形")
   geomPentagon =ogr.CreateGeometryFromWkt("POLYGON ((70 0,85 0,90 15,80 30,65 15,700))")
   oFeaturePentagon.SetGeometry(geomPentagon)
   oLayer.CreateFeature(oFeaturePentagon)
 
   oDS.Destroy()
   print("数据集创建完成!\n")

3.更新

其实更新无非就是获取到field然后设置新值就可以了

其实用SetField()方法就行

import os,sys
from osgeo import gdal
from osgeo import ogr
from osgeo import osr
import numpy
import transformer
# 为了支持中文路径,请添加下面这句代码
 
 
pathname = sys.argv[1]
choose = sys.argv[2]
gdal.SetConfigOption("GDAL_FILENAME_IS_UTF8", "NO")
# 为了使属性表字段支持中文,请添加下面这句
gdal.SetConfigOption("SHAPE_ENCODING", "")
# 注册所有的驱动
ogr.RegisterAll()
# 数据格式的驱动
driver = ogr.GetDriverByName('ESRI Shapefile')
ds = driver.Open(pathname, update=1)
if ds is None:
 print 'Could not open %s'%pathname
 sys.exit(1)
# 获取第0个图层
layer0 = ds.GetLayerByIndex(0);
# 投影
spatialRef = layer0.GetSpatialRef();
# 输出图层中的要素个数
print '要素个数=%d'%(layer0.GetFeatureCount(0))
print '属性表结构信息'
defn = layer0.GetLayerDefn()
fieldindex = defn.GetFieldIndex('x')
xfield = defn.GetFieldDefn(fieldindex)
#新建field
fieldDefn = ogr.FieldDefn('newx', xfield.GetType())
fieldDefn.SetWidth(32)
fieldDefn.SetPrecision(6)
layer0.CreateField(fieldDefn,1)
fieldDefn = ogr.FieldDefn('newy', xfield.GetType())
fieldDefn.SetWidth(32)
fieldDefn.SetPrecision(6)
layer0.CreateField(fieldDefn,1)
feature = layer0.GetNextFeature()
# 下面开始遍历图层中的要素
while feature is not None:
 # 获取要素中的属性表内容
 x = feature.GetFieldAsDouble('x')
 y = feature.GetFieldAsDouble('y')
 newx, newy = transformer.begintrans(choose, x, y)
 feature.SetField('newx', newx)
 feature.SetField('newy', newy)
 layer0.SetFeature(feature)
 feature = layer0.GetNextFeature()
feature.Destroy()
ds.Destroy()

这里我其实想说最重要的是这个SetFeature(),就是你更新好了field的feature一定要重新set一下,不然是根本起不到任何改变的。新建的时候有createfeature,已经设置了,所以不需要set。

网上的教程都是新建和读取,都没有提到这个,结果自己蠢到试了好久都没有发现问题在哪,以为是什么数据类型与设置字段属性不匹配,一头雾水哈哈哈。

补充知识:python使用GDAL生成shp文件

GDAL是一个开源的地理工具包,其支持基本所有的地理操作,其有python、java、c等语言包,是地理信息C端开发不可越过的工具,鉴于python语言的简单性,这里使用python中GDAL包来进行shp文件的生成,这里本质是利用ogc地理标准的坐标字符串来生成shp。

第一步:安装GDAL环境,建议下载后,本地安装,注意与python版本号要对应,可参考网上教程。

第二部:代码分析

引入GDAL工具包

import osgeo.ogr as ogr
import osgeo.osr as osr

注册驱动,这里是ESRI Shapefile类型,并设置shp文件名称

driver = ogr.GetDriverByName("ESRI Shapefile")
data_source = driver.CreateDataSource("ceshi.shp")

注入投影信息,这里使用的4326,表示经纬度坐标,根据情况可以自行更改

srs = osr.SpatialReference()
srs.ImportFromEPSG(4326)

这里定义的是,生成的要素类型,包括点、线、面

#ogr.wkbPoint 点
#ogr.wkbLineString 线
#ogr.wkbMultiPolygon 面

这里的图层名称要与上面注册驱动的shp名称一致

layer = data_source.CreateLayer("ceshi", srs, ogr.wkbLineString)

这里设置要素的属性字段,其中设置了两个字段,分别是Name、data,其中ogr.OFTString表示字符串类型,其长度都是14字节,可自行设置宽度

python使用gdal对shp读取,新建和更新的实例

field_name = ogr.FieldDefn("Name", ogr.OFTString)
field_name.SetWidth(14)
layer.CreateField(field_name)
field_name = ogr.FieldDefn("data", ogr.OFTString)
field_name.SetWidth(14)
layer.CreateField(field_name)

在生成的字段名中插入要素值,即属性表中每行的值

feature = ogr.Feature(layer.GetLayerDefn())
feature.SetField("Name", "ceshi")
feature.SetField("data", "1.2")

核心部分,生成line数据

其中各要素格式如下:

POINT(6 10)
LINESTRING(3 4,10 50,20 25)
POLYGON((1 1,5 1,5 5,1 5,1 1),(2 2,2 3,3 3,3 2,2 2))
MULTIPOINT(3.5 5.6, 4.8 10.5)
MULTILINESTRING((3 4,10 50,20 25),(-5 -8,-10 -8,-15 -4))
MULTIPOLYGON(((1 1,5 1,5 5,1 5,1 1),(2 2,2 3,3 3,3 2,2 2)),((6 3,9 2,9 4,6 3)))
GEOMETRYCOLLECTION(POINT(4 6),LINESTRING(4 6,7 10))
POINT ZM (1 1 5 60)
POINT M (1 1 80)

需要注意的是,这里应该与上面定义的生成要素的类型保持一致,最后是清空缓存,这里多说一句,字符串语法与postgis等开源gis一致,都遵循ogc国际标准

wkt = 'LINESTRING(3 4,10 50,20 25)'
line = ogr.CreateGeometryFromWkt(wkt)
feature.SetGeometry(line)
layer.CreateFeature(feature)

feature = None
data_source = None

结果如下:

python使用gdal对shp读取,新建和更新的实例

用arcgis打开

python使用gdal对shp读取,新建和更新的实例

可以使用该方法,下载在线shp数据,只需要知道所需要素的geojson格式数据中坐标串即可。或者图像识别中获取的矢量边界赋予经纬度。

以上这篇python使用gdal对shp读取,新建和更新的实例就是小编分享给大家的全部内容了,希望能给大家一个参考,也希望大家多多支持三水点靠木。

Python 相关文章推荐
Python字符串切片操作知识详解
Mar 28 Python
Python2.7 实现引入自己写的类方法
Apr 29 Python
利用pyinstaller将py文件打包为exe的方法
May 14 Python
pycharm重置设置,恢复默认设置的方法
Oct 22 Python
python批量获取html内body内容的实例
Jan 02 Python
解决Pyinstaller 打包exe文件 取消dos窗口(黑框框)的问题
Jun 21 Python
Django ORM多对多查询方法(自定义第三张表&ManyToManyField)
Aug 09 Python
Python多线程获取返回值代码实例
Feb 17 Python
基于Python共轭梯度法与最速下降法之间的对比
Apr 02 Python
python爬虫学习笔记之pyquery模块基本用法详解
Apr 09 Python
详解Python中namedtuple的使用
Apr 27 Python
python中关于数据类型的学习笔记
Jul 19 Python
Python实现获取当前目录下文件名代码详解
Mar 10 #Python
python爬虫开发之使用python爬虫库requests,urllib与今日头条搜索功能爬取搜索内容实例
Mar 10 #Python
python+gdal+遥感图像拼接(mosaic)的实例
Mar 10 #Python
python获取栅格点和面值的实现
Mar 10 #Python
Python列表切片常用操作实例解析
Mar 10 #Python
Python numpy多维数组实现原理详解
Mar 10 #Python
python中使用you-get库批量在线下载bilibili视频的教程
Mar 10 #Python
You might like
什么是调频(FM)、调幅(AM)、短波(SW)、长波(LW)
2021/03/01 无线电
随机头像PHP版
2006/10/09 PHP
解决phpmyadmin中文乱码问题。。。
2007/01/18 PHP
php数组函数序列之prev() - 移动数组内部指针到上一个元素的位置,并返回该元素值
2011/10/31 PHP
PHP 将逗号、空格、回车分隔的字符串转换为数组的函数
2012/06/07 PHP
URI、URL和URN之间的区别与联系
2006/12/20 Javascript
jquery animate图片模向滑动示例代码
2011/01/26 Javascript
jquery入门必备的基本认识及实例(整理)
2013/06/24 Javascript
HTML Color Picker(js拾色器效果)
2013/08/27 Javascript
把jQuery的类、插件封装成seajs的模块的方法
2014/03/12 Javascript
JavaScript创建一个object对象并操作对象属性的用法
2015/03/23 Javascript
jQuery幻灯片特效代码分享--鼠标滑过按钮时切换(2)
2020/11/18 Javascript
深入剖析JavaScript面向对象编程
2016/07/12 Javascript
实现easyui的datagrid导出为excel的示例代码
2016/11/10 Javascript
jQuery实现移动端Tab选项卡效果
2017/03/15 Javascript
JS对象与JSON互转换、New Function()、 forEach()、DOM事件流等js开发基础小结
2017/08/10 Javascript
ES6中字符串string常用的新增方法小结
2017/11/07 Javascript
VUE DOM加载后执行自定义事件的方法
2018/09/07 Javascript
JS中如何轻松遍历对象属性的方式总结
2019/08/06 Javascript
Vue中错误图片的处理的实现代码
2019/11/07 Javascript
js中!和!!的区别与用法
2020/05/09 Javascript
jquery实现简单每周轮换的日历
2020/09/10 jQuery
从零学python系列之从文件读取和保存数据
2014/05/23 Python
python 保存float类型的小数的位数方法
2018/10/17 Python
python中的TCP(传输控制协议)用法实例分析
2019/11/15 Python
基于Python正确读取资源文件
2020/09/14 Python
python3从网络摄像机解析mjpeg http流的示例
2020/11/13 Python
HTML5+CSS3实现机器猫
2016/10/17 HTML / CSS
使用spring mvc+localResizeIMG实现HTML5端图片压缩上传的功能
2016/12/16 HTML / CSS
俄罗斯外国汽车和国产汽车配件网上商店:Движком
2020/04/19 全球购物
竞选班干部的演讲稿
2014/04/24 职场文书
环保建议书100字
2014/05/14 职场文书
公务员学习习总书记“三严三实”思想汇报
2014/09/19 职场文书
Redis6.0搭建集群Redis-cluster的方法
2021/05/08 Redis
详解redis在微服务领域的贡献
2021/10/16 Redis
MySQL数据库⾼可⽤HA实现小结
2022/01/22 MySQL