利用python把shp文件写入PostgresSQL数据库

一个有趣的灵魂W

目的：

把shp文件写入pg数据库。注意：shp文件含点、线、面（‘POINT’，'LINESTRING'，'POLYGON'）~这是重点。

材料：

PG数据库（postgreSQL），python，postgis（重点）

过程：

Tip~首先，我想先说一下在python中运行出现的报错问题（如果你们遇到这个问题可以看一眼，如果你没遇到这个问题，就跳过这段吧）：

这个问题形成的原因，是没有安装PostGIS，最终导致地理信息无法正常输入。一旦规避掉这个问题，那把数据输入到PG数据库就不是什么难事了。

不管怎么样，你需要安装好Python和PostresSQL数据库~这里不加展开了，然后你需要安装~~~~~~PostGIS~~~~~~由PostGIS官网下载：

http://www.postgis.org/

选择对应的的pg数据库版本(我的pg数据库是pg11)：

https://winnie.postgis.net/download/windows/pg11/buildbot/

windows一般选择exe版下载（我的操作系统是windows）：

安装PostGIS软件的时候需要选择pg数据库的安装根目录进行对应安装（我的pg数据库路径是：C:Program Filespostgresql）。安装的时候设置好用户密码还有数据库~（这部分你要会一些基本的pg数据库~可以百度）。准备就绪后进入pg数据库的sql界面，输入：

CREATE EXTENSION postgis;

（这步可以忽略，如果python运行出错，回头运行也可以）

确认PosrGIS（如下图）

首先，你需要有一个shp文件。我有一个点文件。

其次，将shp文件导入Geodataframe中。

再把Geodataframe写入pg数据库。

你需要导入这几个第三方库：

import geopandas as gpd
import psycopg2
from sqlalchemy import create_engine
from geoalchemy2 import Geometry,WKTElement

接着：

engine = create_engine('postgresql://abc:password@localhost:port/databasename')
#abc:password对应代表你的用户名和密码，格式为用户名：密码
#localhost:port就不解释了

map_data = gpd.GeoDataFrame.from_file('D:/Thesis/fjsouthwest/New_Shapefile.shp')
map_data['geometry'] = map_data['geometry'].apply(lambda x: WKTElement(x.wkt,4326))
map_data.to_sql("tsea",engine,if_exists='replace',index=False,
                dtype={
                'geometry': Geometry(geometry_type ='POINT', srid=4326)}
                )
#"tsea"是表名
#dtype中的'geometry'代表具体字段，相当于'geometry'字段为地理信息

终于，顺利写入（鬼知道我在报错的过程中经历了什么）：

更多的，利用GDAL写入shp也可以正常运行了：

import os
os.system('ogr2ogr '+'-overwrite '+'-f '+'"'+"PostgreSQL"+'"'+' PG:'+'"'+"host=localhost user=user dbname=dbname password=password"+'"'+' '+'"'+"D:/Thesis/fjsouthwest/grid/mxn/mxn.shp"+'"')

往期

Python-gdal离线文档下载

百度网盘下载很慢吧，Pandownload之后的新工具-亿寻

PyCharm2019亲测破解方式

中国范围2019年道路网数据分享

分享一套中国区域的矢量图层（到县级）

微信号：一个有趣的灵魂W

关注我们，了解更多