Python+GIS — shapefile转geojson(核心代码只需2行)

背景

  geojson是地图可视化系统中最常用的地理数据格式,几乎所有主流地图可视化库或框架都支持geojson数据的加载。geojson数据通常是由其他数据转换而来的,最常见的就是ESRI Shpfile数据转geojson,专业GIS软件中诸如QGIS,FME等都支持二者的互相转换,但这些软件不仅过于庞大,而且专业性较强,作为一个GIS专业的我也不愿使用。之前用java写过二者转换的工具,但代码实在过于繁琐。直到使用了python的geopandas库。

步骤

1. 安装geopandas

最简单的方式:

conda install geopandas

也可以使用pip进行安装,但需要把其他的一些依赖库事先安装好(numpy, pandas, shapely, fiona, pyproj等),这个不做赘述。

2. 代码

import geopandas as gpd

def shp2gj(input_file, output_file):
    data = gpd.read_file(input_file)
    data.to_file(output_file, driver="GeoJSON", encoding='utf-8') # 指定utf-8编码,防止中文乱码
    print('Success: File '+input_file.split('\\') [-1] + ' conversion completed')

两个参数,input_file(输入的文件,xxx.shp),output_file(输出的文件,xxx.json)。主要代码只有两行:读数据(read_file),写数据(to_file);写数据时指定 driver="GeoJSON" 即可;为防止中文乱码,指定 encoding='utf-8'

3. 扩展

  上述代码将shapefile转为geojson。同样也可以将shapefile或geojson转为其他格式,如:将geojson转为shapefile,只需把driver参数改为ESRI Shapefile即可。

data = gpd.read_file(r'yourPath\xxx.json')
data.to_file(r'yourPath\xxx.shp', driver="ESRI Shapefile", encoding='utf-8') # 指定utf-8编码,防止中文乱码

有时geojson没有空间参考,需要指定空间参考:

import geopandas as gpd
from fiona.crs import from_epsg

data = gpd.read_file(r'yourPath\xxx.josn')
data.crs = from_epsg(4326) # 指定空间参考为4326(WGS84坐标)
data.to_file(r'yourPath\xxx.shp', driver="ESRI Shapefile", encoding='utf-8')

driver支持的数据类型可通过如下代码查看:

>>> import fiona
>>> fiona.supported_drivers
{'ESRI Shapefile': 'raw', 'ARCGEN': 'r', 'PCIDSK': 'r', 'SUA': 'r', 
'DGN': 'raw', 'SEGY': 'r', 'MapInfo File': 'raw', 'GeoJSON': 'rw', 'PDS': 'r', 
'FileGDB': 'raw', 'GPX': 'raw', 'DXF': 'raw', 'GMT': 'raw', 'Idrisi': 'r', 
'GPKG': 'rw', 'OpenFileGDB': 'r', 'BNA': 'raw', 'AeronavFAA': 'r', 
'GPSTrackMaker': 'raw'}

你可能感兴趣的:(gis,python)