python创建shape

import shapefile
import json
import os

#shapefile="polygon.shp";
#jsonfile="社区网格.json";

def run():
    data=readJSON()
    datalist=data["list"]
    file=shapefile.Writer(target="社区网格.shp",shapeType=shapefile.POLYGON,autoBalance=True);
    #设置属性信息
    file.field('nere', 'C', '40') #’SECOND_FLD’为字段名称,C代表数据类型为字符串,长度为40
    file.field('fileName', 'C', '40')
    file.field('url', 'C', '40')
    file.field('photo', 'C', '40')
    file.field('netGrnereNum', 'C', '40')
    file.field('color', 'C', '40')
    file.field('num', 'C', '40')
    file.field('lat', 'C', '40')
    file.field('lon', 'C', '40')
    file.field('togriga', 'C', '40')
    file.field('gudumji', 'C', '40')
    file.field('memo', 'C', '40')

    #写入数据
    for item in datalist:
        nere=item["nere"]
        fileName = item["fileName"]
        url = item["url"]
        photo = item["photo"]
        netGrnereNum = item["netGrnereNum"]
        color = item["color"]
        num = item["num"]
        lat = item["lat"]
        lon = item["lon"]
        togriga = item["togriga"]
        gudumji = item["gudumji"]
        memo = item["memo"]
        polysStr = item["hotinfo"]
        polys=parsing(polysStr);
        #写入
        file.poly(polys=polys)
        file.record(nere,fileName,url,photo,netGrnereNum,color,num,lat,lon,togriga,gudumji,memo)
        print("-------写入成功---------")
    #保存文件
    file.save("社区网格.shp")

#读取json文件
def readJSON():
    f = open("社区网格.json", "r",encoding="utf-8")  # 设置文件对象
    _str = f.read()  # 将txt文件的所有内容读入到字符串str中
    f.close()  # 关闭文件
    print("读取json完成")
    if _str.startswith(u'\ufeff'):
        _str = _str.encode('utf8')[3:].decode('utf8')
    data = json.loads(_str,encoding='utf8')
    return data;
    #print(data)

#数据解析
def parsing(text):
    data=[]
    strs=text.split(",");
    for line in strs:
        xy=line.split(" ");
        x_y=[float(xy[0]),float(xy[1])]
        data.append(x_y)
    _data=[];
    _data.append(data)
    return _data;


if __name__=="__main__":
    print("执行程序!!!");
    run();
    print("程序结束!!!");

 

import os

try:
    from osgeo import ogr, gdal
    from osgeo import osr
except:
    import ogr
# 属性
reader = [
    {
        'Name': "a",
        'Region': "我是",
        'Latitude': "86478.950188794",
        'Longitude': "74905.3874084952",
        'Elev': 10,

    }, {
        'Name': "b",
        'Region': "b",
        'Latitude': "86480",
        'Longitude': "74915",
        'Elev': 20,

    }
]
gdal.SetConfigOption("GDAL_FILENAME_IS_UTF8", "YES")  # 路径中文
gdal.SetConfigOption("SHAPE_ENCODING", "GBK")  # 属性中文
"""
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)
POINT EMPTY
MULTIPOLYGON EMPTY"""


def createPoint(reader):
    # 设置shapefile驱动
    driver = ogr.GetDriverByName("ESRI Shapefile")

    # 创建数据源
    data_source = driver.CreateDataSource("point.shp")

    # 创建空间参考,WGS84
    srs = osr.SpatialReference()
    srs.ImportFromEPSG(4326)

    # 创建图层
    # layer = data_source.CreateLayer("point", srs,ogr.wkbPoint) # 带坐标系文件

    layer = data_source.CreateLayer("point", geom_type=ogr.wkbPoint)  # 无坐标系文件

    # 设置字段
    field_name = ogr.FieldDefn("名称", ogr.OFTString)  # ogr.OFTString -> shapefile String
    field_name.SetWidth(24)  # 设置长度
    layer.CreateField(field_name)  # 创建字段
    field_region = ogr.FieldDefn("Region", ogr.OFTString)
    field_region.SetWidth(24)
    layer.CreateField(field_region)  # 创建字段
    layer.CreateField(
        ogr.FieldDefn("Latitude", ogr.OFTReal))  # 创建 名为Latitude 的字段数据类型为 Double 23/15 . ogr.OFTReal->shapefile Double
    layer.CreateField(ogr.FieldDefn("Longitude", ogr.OFTReal))
    layer.CreateField(
        ogr.FieldDefn("Elevation", ogr.OFTInteger))  # 创建 名为 Elevation 的字段数据类型为 long 9 ogr.OFTInteger ->shapefile Long

    for row in reader:
        # 创建特征
        feature = ogr.Feature(layer.GetLayerDefn())
        # 和设置字段内容进行关联  ,从数据源中写入数据
        feature.SetField("名称", row['Name'])
        feature.SetField("Region", row['Region'])
        feature.SetField("Latitude", row['Latitude'])
        feature.SetField("Longitude", row['Longitude'])
        feature.SetField("Elevation", row['Elev'])

        # 创建WKT 文本点
        wkt = "POINT(%f %f)" % (
            float(row['Longitude']), float(row['Latitude']))

        # 生成实体点
        point = ogr.CreateGeometryFromWkt(wkt)

        # 使用点
        feature.SetGeometry(point)
        # 添加点
        layer.CreateFeature(feature)
        # 关闭 特征
        feature = None
    # 关闭数据
    data_source = None


def createLine(reader):
    # 设置shapefile驱动
    driver = ogr.GetDriverByName("ESRI Shapefile")

    # 创建数据源
    data_source = driver.CreateDataSource("line.shp")

    # 创建空间参考,WGS84
    srs = osr.SpatialReference()
    srs.ImportFromEPSG(4326)

    # 创建图层
    layer = data_source.CreateLayer("line", srs, ogr.wkbLineString)

    # 设置字段
    field_name = ogr.FieldDefn("名称", ogr.OFTString)  # ogr.OFTString -> shapefile String
    field_name.SetWidth(24)  # 设置长度
    layer.CreateField(field_name)  # 创建字段
    field_region = ogr.FieldDefn("Region", ogr.OFTString)
    field_region.SetWidth(24)
    layer.CreateField(field_region)  # 创建字段
    layer.CreateField(
        ogr.FieldDefn("Latitude", ogr.OFTReal))  # 创建 名为Latitude 的字段数据类型为 Double 23/15 . ogr.OFTReal->shapefile Double
    layer.CreateField(ogr.FieldDefn("Longitude", ogr.OFTReal))
    layer.CreateField(
        ogr.FieldDefn("Elevation", ogr.OFTInteger))  # 创建 名为 Elevation 的字段数据类型为 long 9 ogr.OFTInteger ->shapefile Long

    for row in reader:
        # 创建特征
        feature = ogr.Feature(layer.GetLayerDefn())
        # 和设置字段内容进行关联  ,从数据源中写入数据
        feature.SetField("名称", row['Name'])
        feature.SetField("Region", row['Region'])
        feature.SetField("Latitude", row['Latitude'])
        feature.SetField("Longitude", row['Longitude'])
        feature.SetField("Elevation", row['Elev'])

        # 创建WKT 文本点
        wkt = "LINESTRING(%f %f,%f %f)" % (
            float(row['Longitude']), float(row['Latitude']), float(row['Latitude']), float(row['Latitude']))

        # 生成实体点
        point = ogr.CreateGeometryFromWkt(wkt)

        # 使用点
        feature.SetGeometry(point)
        # 添加点
        layer.CreateFeature(feature)
        # 关闭 特征
        feature = None
    # 关闭数据
    data_source = None


def createPolygon(reader):
    # 设置shapefile驱动
    driver = ogr.GetDriverByName("ESRI Shapefile")

    # 创建数据源
    data_source = driver.CreateDataSource("polygon.shp")

    # 创建空间参考,WGS84
    srs = osr.SpatialReference()
    srs.ImportFromEPSG(4326)

    # 创建图层
    layer = data_source.CreateLayer("polygon", srs, ogr.wkbPolygon)

    # 设置字段
    field_name = ogr.FieldDefn("名称", ogr.OFTString)  # ogr.OFTString -> shapefile String
    field_name.SetWidth(24)  # 设置长度
    layer.CreateField(field_name)  # 创建字段
    field_region = ogr.FieldDefn("Region", ogr.OFTString)
    field_region.SetWidth(24)
    layer.CreateField(field_region)  # 创建字段
    layer.CreateField(
        ogr.FieldDefn("Latitude", ogr.OFTReal))  # 创建 名为Latitude 的字段数据类型为 Double 23/15 . ogr.OFTReal->shapefile Double
    layer.CreateField(ogr.FieldDefn("Longitude", ogr.OFTReal))
    layer.CreateField(
        ogr.FieldDefn("Elevation", ogr.OFTInteger))  # 创建 名为 Elevation 的字段数据类型为 long 9 ogr.OFTInteger ->shapefile Long

    for row in reader:
        # 创建特征
        feature = ogr.Feature(layer.GetLayerDefn())
        # 和设置字段内容进行关联  ,从数据源中写入数据
        feature.SetField("名称", row['Name'])
        feature.SetField("Region", row['Region'])
        feature.SetField("Latitude", row['Latitude'])
        feature.SetField("Longitude", row['Longitude'])
        feature.SetField("Elevation", row['Elev'])

        # 创建WKT 文本面
        wkt = "POLYGON((120 30,30 120 ,120 120,120 30))"

        # 生成实体点
        point = ogr.CreateGeometryFromWkt(wkt)

        # 使用点
        feature.SetGeometry(point)
        # 添加点
        layer.CreateFeature(feature)
        # 关闭 特征
        feature = None
    # 关闭数据
    data_source = None


if __name__ == '__main__':
    # createLine(reader)
    createPoint(reader)
    # createPolygon(reader)
    pass

 

posted @ 2018-12-19 17:05  扰扰  阅读(1610)  评论(0编辑  收藏  举报