Python PostGIS 创建表|极客教程

Python PostGIS 创建表

PostGIS 是一个开源的地理信息系统扩展，它使得 PostgreSQL 数据库具备了处理地理空间数据的能力。在本文中，我们将介绍如何使用 Python 操作 PostGIS 创建表格并插入数据。

环境准备

首先，我们需要安装 psycopg2 和 sqlalchemy 这两个 Python 库。可通过以下命令安装：

pip install psycopg2 sqlalchemy

接下来，我们需要确保 PostgreSQL 数据库中已经安装了 PostGIS 扩展。如果没有安装，请按照官方文档进行安装。

创建数据库连接

我们首先需要建立数据库连接，并创建一个表来存储地点数据。以下是一个简单的 Python 脚本来连接到 PostgreSQL 数据库并创建一个名为 locations 的表格：

import psycopg2
from sqlalchemy import create_engine

# 数据库连接信息
DB_HOST = 'localhost'
DB_NAME = 'mydatabase'
DB_USER = 'myuser'
DB_PASSWORD = 'mypassword'

# 创建数据库连接
engine = create_engine(f'postgresql+psycopg2://{DB_USER}:{DB_PASSWORD}@{DB_HOST}/{DB_NAME}')
connection = engine.connect()

# 创建表格
connection.execute('CREATE TABLE locations (id SERIAL PRIMARY KEY, name VARCHAR, geom GEOMETRY(Point));')

# 关闭连接
connection.close()

上面的代码中，我们首先使用 psycopg2 库和 sqlalchemy 库建立了一个数据库连接。然后我们执行创建表格的 SQL 语句，其中包括了一个 id 字段作为主键、一个 name 字段用来存储地点名称、一个 geom 字段用来存储地点的几何信息。

插入数据

接下来，我们将向 locations 表格中插入一些地点数据：

from shapely.geometry import Point
from geoalchemy2 import WKTElement

# 创建一个点对象
point = Point(0, 0)
point_wkt = WKTElement(point.wkt, srid=4326) 

# 插入数据
connection = engine.connect()
connection.execute("INSERT INTO locations (name, geom) VALUES ('Origin', ST_GeomFromText(:point, 4326));", point=point_wkt)
connection.close()

上面的代码中，我们使用 shapely 库创建了一个 Point 对象表示原点，然后使用 geoalchemy2 库的 WKTElement 函数将这个点对象转换为几何文本。最后，我们执行插入数据的 SQL 语句将原点数据插入到 locations 表格中。

查询数据

最后，我们可以查询 locations 表格中的数据：

connection = engine.connect()
result = connection.execute('SELECT * FROM locations;')
for row in result:
    print(row)
connection.close()

以上代码会输出如下结果：

(1, 'Origin', 'POINT(0 0)')

这样我们就成功地使用 Python 操作 PostGIS 创建了表格并插入了数据。

Python PostGIS 创建表