import os
import gdal
import numpy as np
from osgeo import ogr
from geopandas import *
#读取矢量数据元数据信息(要素包括 矩形坐标,字段及类型等)
vdata_dir='C:\\Users\\admin\\Desktop\\空间信息处理与系统开发资料\\第七周\\Lesson7\\2-Second'
filename=os.path.join(vdata_dir,'FirstLayer.shp')
#print(filename)
ds=ogr.Open(filename,0) #只读
layer=ds.GetLayer(0)
l,r,b,t=layer.GetExtent() #边界
#print(l,r,b,t)
#读取整个图层的feature(需要geopandas包)
shpdata=GeoDataFrame.from_file(filename)#从文件中加载可以被fiona识别的任何格式的一个GeoDataFrame
features=[shpdata.geometry.__geo_interface__]#获取矢量数据的feature
#print(features)
#新建一个空的栅格数据
cols=int(r-l)
rows=int(t-b)
#print(cols,rows)
#创建空栅格数据
out_raster=os.path.join('C:\\Users\\admin\\Desktop\\空间信息处理与系统开发资料\\第七周\\Lesson7\\out_put','vectorToRaster.tif')
driver = gdal.GetDriverByName('GTiff')
dataset_out = driver.Create(out_raster, cols, rows, 1, gdal.GDT_Float32)
#添加图的仿射信息
nodata_value=1
geo_transform=np.zeros(6)#仿射变换6个参数
#将仿射变换参数和nodata值设置到栅格数据集中
dataset_out.SetGeoTransform(geo_transform)
dataset_out.GetRasterBand(1).SetNoDataValue(nodata_value)
dataset_out.GetRasterBand(1).FlushCache()
#调用GDAL栅格化函数进行矢量转栅格运算
gdal.RasterizeLayer(dataset_out,[1],layer,options=['ATTRIBUTE=value']) #四个参数分别为:目标栅格数据集、第几波段、矢量数据、栅格像素值为矢量value字段值
print(dataset_out.ReadAsArray(0,0,cols,rows))
#释放文件句柄
del dataset_out
del ds
python 矢量转栅格
最新推荐文章于 2024-06-01 09:01:15 发布