The common network coordinate system conversion of shp files is realized by python+gdal! 2020-12-11

When installing gdal, pay attention to the corresponding python version. The download address of the gdal library file (search: Geospatial Data Abstraction Library): https://www.lfd.uci.edu/~gohlke/pythonlibs/#gdal

gdal python api reference document address: https://gdal.org/python/index.html

 

Two py files, one coorTrans.py is the algorithm file, and one shpCoordTransform.py is the function realization part. It supports the mutual conversion of these three coordinate systems: WGS84, GCJ02, BD09 (Sky Map, Google Map/High German Map, Baidu Map) .

Folder where coorTrans.py is located: E:\python, algorithm code:

#!/usr/bin/env python
# -*- coding:utf-8 -*-
# @File  : 坐标转换.py
# @Author: hukelin
# @Date  : 2019/7/18
# @Desc  :hello world!
'''
百度坐标转换
'''
import math

x_pi = 3.14159265358979324 * 3000.0 / 180.0
pi = 3.1415926535897932384626  # π
a = 6378245.0  # 长半轴
ee = 0.00669342162296594323  # 偏心率平方
coordinate = []
lng = []
lat = []
converted_lng = []
converted_lat = []

def gcj02_to_bd09(lng, lat):
    """
    火星坐标系(GCJ-02)转百度坐标系(BD-09)
    谷歌、高德——>百度
    :param lng:火星坐标经度
    :param lat:火星坐标纬度
    :return:
    """
    z = math.sqrt(lng * lng + lat * lat) + 0.00002 * math.sin(lat * x_pi)
    theta = math.atan2(lat, lng) + 0.000003 * math.cos(lng * x_pi)
    bd_lng = z * math.cos(theta) + 0.0065
    bd_lat = z * math.sin(theta) + 0.006
    return [bd_lng, bd_lat]


def bd09_to_gcj02(bd_lon, bd_lat):
    """
    百度坐标系(BD-09)转火星坐标系(GCJ-02)
    百度——>谷歌、高德
    :param bd_lat:百度坐标纬度
    :param bd_lon:百度坐标经度
    :return:转换后的坐标列表形式
    """
    x = bd_lon - 0.0065
    y = bd_lat - 0.006
    z = math.sqrt(x * x + y * y) - 0.00002 * math.sin(y * x_pi)
    theta = math.atan2(y, x) - 0.000003 * math.cos(x * x_pi)
    gg_lng = z * math.cos(theta)
    gg_lat = z * math.sin(theta)
    return [gg_lng, gg_lat]


def wgs84_to_gcj02(lng, lat):
    """
    WGS84转GCJ02(火星坐标系)
    :param lng:WGS84坐标系的经度
    :param lat:WGS84坐标系的纬度
    :return:
    """
    if out_of_china(lng, lat):  # 判断是否在国内
        return [lng, lat]
    dlat = _transformlat(lng - 105.0, lat - 35.0)
    dlng = _transformlng(lng - 105.0, lat - 35.0)
    radlat = lat / 180.0 * pi
    magic = math.sin(radlat)
    magic = 1 - ee * magic * magic
    sqrtmagic = math.sqrt(magic)
    dlat = (dlat * 180.0) / ((a * (1 - ee)) / (magic * sqrtmagic) * pi)
    dlng = (dlng * 180.0) / (a / sqrtmagic * math.cos(radlat) * pi)
    mglat = lat + dlat
    mglng = lng + dlng
    return [mglng, mglat]


def gcj02_to_wgs84(lng, lat):
    """
    GCJ02(火星坐标系)转GPS84
    :param lng:火星坐标系的经度
    :param lat:火星坐标系纬度
    :return:
    """
    if out_of_china(lng, lat):
        return [lng, lat]
    dlat = _transformlat(lng - 105.0, lat - 35.0)
    dlng = _transformlng(lng - 105.0, lat - 35.0)
    radlat = lat / 180.0 * pi
    magic = math.sin(radlat)
    magic = 1 - ee * magic * magic
    sqrtmagic = math.sqrt(magic)
    dlat = (dlat * 180.0) / ((a * (1 - ee)) / (magic * sqrtmagic) * pi)
    dlng = (dlng * 180.0) / (a / sqrtmagic * math.cos(radlat) * pi)
    mglat = lat + dlat
    mglng = lng + dlng
    return [lng * 2 - mglng, lat * 2 - mglat]


def bd09_to_wgs84(bd_lon, bd_lat):
    lon, lat = bd09_to_gcj02(bd_lon, bd_lat)
    return gcj02_to_wgs84(lon, lat)


def wgs84_to_bd09(lon, lat):
    lon, lat = wgs84_to_gcj02(lon, lat)
    return gcj02_to_bd09(lon, lat)


def _transformlat(lng, lat):
    ret = -100.0 + 2.0 * lng + 3.0 * lat + 0.2 * lat * lat + \
          0.1 * lng * lat + 0.2 * math.sqrt(math.fabs(lng))
    ret += (20.0 * math.sin(6.0 * lng * pi) + 20.0 *
            math.sin(2.0 * lng * pi)) * 2.0 / 3.0
    ret += (20.0 * math.sin(lat * pi) + 40.0 *
            math.sin(lat / 3.0 * pi)) * 2.0 / 3.0
    ret += (160.0 * math.sin(lat / 12.0 * pi) + 320 *
            math.sin(lat * pi / 30.0)) * 2.0 / 3.0
    return ret


def _transformlng(lng, lat):
    ret = 300.0 + lng + 2.0 * lat + 0.1 * lng * lng + \
          0.1 * lng * lat + 0.1 * math.sqrt(math.fabs(lng))
    ret += (20.0 * math.sin(6.0 * lng * pi) + 20.0 *
            math.sin(2.0 * lng * pi)) * 2.0 / 3.0
    ret += (20.0 * math.sin(lng * pi) + 40.0 *
            math.sin(lng / 3.0 * pi)) * 2.0 / 3.0
    ret += (150.0 * math.sin(lng / 12.0 * pi) + 300.0 *
            math.sin(lng / 30.0 * pi)) * 2.0 / 3.0
    return ret


def out_of_china(lng, lat):
    """
    判断是否在国内,不在国内不做偏移
    :param lng:
    :param lat:
    :return:
    """
    return not (lng > 73.66 and lng < 135.05 and lat > 3.86 and lat < 53.55)

The folder E:\python where shpCoordTransform.py is located, the coordinate conversion code:

import os, sys
from osgeo import gdal
from osgeo import ogr
from osgeo import osr
import json
import numpy
# import transformer
from osgeo._ogr import CreateGeometryFromJson

sys.path.append(r'E:\python')
import coorTrans

# 为了支持中文路径,请添加下面这句代码
# pathname = sys.argv[1]
# choose = sys.argv[2]
def transformXY(pathname,type2type):
    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))
    feature = layer0.GetNextFeature()
    geoType = layer0.GetGeomType()
    # 下面开始遍历图层中的要素
    while feature is not None:
        # 获取要素中的属性表内容
        geom = feature.GetGeometryRef()
        if geoType==1:
            xy = geom.GetPoint()
            x2,y2 = transformByType(xy[0],xy[1],type2type)
            geom.Empty()
            geom.SetPoint(0,x2,y2)
        elif geoType==2:
            pts = geom.GetPoints()
            for i in range(len(pts)):
                pt = pts[i]
                x2,y2 = transformByType(pt[0], pt[1], type2type)
                geom.SetPoint(i,x2,y2)
        elif geoType==3:
            for geomgeom in geom:
                pts = geomgeom.GetPoints()
                for i in range(len(pts)):
                    pt = pts[i]
                    x2,y2 = transformByType(pt[0], pt[1], type2type)
                    geomgeom.SetPoint(i,x2,y2)

        layer0.SetFeature(feature)
        feature = layer0.GetNextFeature()
    # feature.Destroy()
    ds.Destroy()

def transformByType(x,y,type):
    if type=='bd09->gcj02':
        return coorTrans.bd09_to_gcj02(x, y)
    elif type=='bd09->2000':
        return coorTrans.bd09_to_wgs84(x, y)
    elif type=='gcj02->bd09':
        return coorTrans.gcj02_to_bd09(x, y)
    elif type=='gcj02->2000':
        return coorTrans.gcj02_to_wgs84(x, y)
    elif type=='2000->bd09':
        return coorTrans.wgs84_to_bd09(x, y)
    elif type=='2000->gcj02':
        return coorTrans.wgs84_to_gcj02(x, y)
    return x,y

if __name__ == "__main__":
    try:
        pathname1 = r'E:\temp\point.shp'
        pathname2 = r'E:\temp\line.shp'
        pathname3 = r'E:\temp\polygon.shp'
        type = 'bd09->gcj02'
        transformXY(pathname1, type)
        transformXY(pathname2, type)
        transformXY(pathname3, type)
    except ValueError:
        print(ValueError)

 

Guess you like

Origin blog.csdn.net/qq503690160/article/details/111043782
Recommended