Python+GDAL | 读取矢量并写出txt

最新推荐文章于 2024-07-09 09:42:52 发布

许雪爽

最新推荐文章于 2024-07-09 09:42:52 发布

阅读量690

点赞数 1

分类专栏： Python+GDAL 文章标签： python gdal ogr 矢量

Python+GDAL 专栏收录该内容

6 篇文章 5 订阅

订阅专栏

这篇文章主要描述了如何使用GDAL/OGR打开矢量文件、读取属性表，并将部分属性写出至txt。

代码

import ogr,sys,os
import numpy as np

os.chdir(r'E:\')

#设置driver,并打开矢量文件
driver = ogr.GetDriverByName('ESRI Shapefile')
ds = driver.Open('sites.shp', 0)
if ds is None:
    print("Could not open", 'sites.shp')
    sys.exit(1)
#获取图册
layer = ds.GetLayer()

#要素数量
numFeatures = layer.GetFeatureCount()
print("Feature count: "+str(numFeatures))

#获取范围
extent = layer.GetExtent()
print("Extent:", extent)
print("UL:", extent[0],extent[3])
print("LR:", extent[1],extent[2])

#获取要素
feature = layer.GetNextFeature()
ids = []
xs = []
ys = []
covers = []
#循环每个要素属性
while feature:
    #获取字段“id”的属性
    id = feature.GetField('id')
    #获取空间属性
    geometry = feature.GetGeometryRef()
    x = geometry.GetX()
    y = geometry.GetY()
    cover = feature.GetField('cover')
    ids.append(id)
    xs.append(x)
    ys.append(y)
    covers.append(cover)
    feature = layer.GetNextFeature()

data = [ids, xs, ys, covers]
data = np.array(data)
data = data.transpose()

#写出致txt
np.savetxt('myfile.txt',data, fmt='%s %s %s %s')
np.savetxt('myfile.csv',data, fmt='%s %s %s %s')

#释放文件空间
layer.ResetReading()
feature.Destroy()
ds.Destroy()