矢量数据处理（python、gdal）

纯粹小白

已于 2023-09-13 17:19:43 修改

阅读量237

点赞数

文章标签： python 开发语言

于 2023-09-13 17:18:05 首次发布

本文链接：https://blog.csdn.net/weixin_44214236/article/details/132858967

版权

文章目录

- - 1.创建矢量数据

1.创建矢量数据

	EPSG：[https://epsg.io/]

（1）创建点数据

from osgeo import ogr, osr
import os

def create_point_shp(path, file, coordinates, x, y):
    '''
    创建矢量点数据
    :param path: file所在文件夹
    :param file: 文件名
    :param coordinates: 空间参考：EPSG（例如：4326表示WGS84，可在"https://epsg.io/"查找）
    :param x:x轴坐标
    :param y:y轴坐标
    :return:
    '''
    driver = ogr.GetDriverByName("ESRI Shapefile")
    data_source = driver.CreateDataSource(os.path.join(path, file))  # 设置shp文件名称
    # 定义空间坐标系（epsg）
    srs = osr.SpatialReference()
    srs.ImportFromEPSG(coordinates)
    layer = data_source.CreateLayer(file[:-4], srs, ogr.wkbPoint)  # 图层名称要与shp名称一致
    field_name = ogr.FieldDefn("Name", ogr.OFTString)  ## 设置属性
    field_name.SetWidth(20)  ## 设置长度
    layer.CreateField(field_name)  ## 创建字段
    field_Longitude = ogr.FieldDefn("Longitude", ogr.OFTReal)  ## 设置属性
    layer.CreateField(field_Longitude)  ## 创建字段
    field_Latitude = ogr.FieldDefn("Latitude", ogr.OFTReal)  ## 设置属性
    layer.CreateField(field_Latitude)  ## 创建字段
    feature = ogr.Feature(layer.GetLayerDefn())
    feature.SetField("Name", "point")  ## 设置字段值
    feature.SetField("Longitude", str(x))  ## 设置字段值
    feature.SetField("Latitude", str(y))  ## 设置字段值
    wkt = "POINT(%f %f)" % (float(x), float(y))  ## 创建点
    point = ogr.CreateGeometryFromWkt(wkt)  ## 生成点
    feature.SetGeometry(point)  ## 设置点
    layer.CreateFeature(feature)  ## 添加点
    feature = None  ## 关闭属性
    data_source = None  ## 关闭数据
    
    
if __name__ == "__name__":
    create_point_shp("./data", "points.shp", 4326, 113, 40.9)

（2）创建线数据

from osgeo import ogr,osr
 
def create_line():
    ## 生成线矢量文件 ##
    driver = ogr.GetDriverByName("ESRI Shapefile")
    data_source = driver.CreateDataSource("Line.shp") ## shp文件名称
    srs = osr.SpatialReference()
    srs.ImportFromEPSG(4326) ## 空间参考：WGS84
    layer = data_source.CreateLayer("Line", srs, ogr.wkbLineString) ## 图层名称要与shp名称一致
    field_name = ogr.FieldDefn("Name", ogr.OFTString) ## 设置属性
    field_name.SetWidth(20)  ## 设置长度
    layer.CreateField(field_name)  ## 创建字段
    field_length = ogr.FieldDefn("Length", ogr.OFTReal)  ## 设置属性
    layer.CreateField(field_length)  ## 创建字段
    feature = ogr.Feature(layer.GetLayerDefn())
    feature.SetField("Name", "line")  ## 设置字段值
    feature.SetField("Length", "100")  ## 设置字段值
    wkt = 'LINESTRING(116.4 39.9, 116.42 39.91, 116.39 39.92, 116.38 39.91)' ## 创建线
    line = ogr.CreateGeometryFromWkt(wkt) ## 生成线
    feature.SetGeometry(line)  ## 设置线
    layer.CreateFeature(feature)  ## 添加线
    feature = None ## 关闭属性
    data_source = None ## 关闭数据

（3）创建面数据

from osgeo import ogr,osr
 
def create_polygon():
    ## 生成线矢量文件 ##
    driver = ogr.GetDriverByName("ESRI Shapefile")
    data_source = driver.CreateDataSource("Polygon.shp") ## shp文件名称
    srs = osr.SpatialReference()
    srs.ImportFromEPSG(4326) ## 空间参考：WGS84
    layer = data_source.CreateLayer("Polygon", srs, ogr.wkbPolygon) ## 图层名称要与shp名称一致
    field_name = ogr.FieldDefn("Name", ogr.OFTString) ## 设置属性
    field_name.SetWidth(20)  ## 设置长度
    layer.CreateField(field_name)  ## 创建字段
    field_length = ogr.FieldDefn("Area", ogr.OFTReal)  ## 设置属性
    layer.CreateField(field_length)  ## 创建字段
    feature = ogr.Feature(layer.GetLayerDefn())
    feature.SetField("Name", "polygon")  ## 设置字段值
    feature.SetField("Area", "500")  ## 设置字段值
    wkt = "POLYGON((116.41 39.89, 116.41 39.91, 116.39 39.91, 116.39 39.89))" ## 创建面
    polygon = ogr.CreateGeometryFromWkt(wkt) ## 生成面
    feature.SetGeometry(polygon)  ## 设置面
    layer.CreateFeature(feature)  ## 添加面
    feature = None ## 关闭属性
    data_source = None ## 关闭数据