Вставьте точку в PostGIS, используя Python

Ответы:

21

Сначала установите пакет psycopg2 , интерфейс Pythonic для PostgreSQL.

Затем используйте ST_MakePoint:

>>> import psycopg2
>>> conn = psycopg2.connect(dbname=..., port=..., user=...,
                            password=..., host=...)
>>> cur = conn.cursor()
>>> x, y, z, = 32, 34, 0
>>> cur.execute("SELECT ST_SetSRID(ST_MakePoint(%s, %s, %s),4326);", (x, y, z))
>>> cur.fetchall()
[('01010000A0E6100000000000000000404000000000000041400000000000000000',)]

ST_AsText может использоваться для проверки значений:

>>> cur.execute("SELECT ST_AsText(ST_SetSRID(ST_MakePoint(%s, %s, %s),4326));", (x, y, z))
>>> cur.fetchall()
[('POINT Z (32 34 0)',)]

Заметки

  • Помните , что (lat, lon)это (y, x)не (x, y).
  • Всегда используйте параметры, а не строковые манипуляции, чтобы предотвратить внедрение SQL . В этих примерах мы ставим кортеж (x, y, z)в конце, чтобы psycopg2справиться с заменой.
Адам Матан
источник
20

Для более сложных геометрий, таких как геометрия LineString и Polygon, вы можете обрабатывать их с помощью Shapely, а затем передавать их через psycopg2 как закодированный в шестнадцатеричном формате WKB. Обратите внимание, что Shapely 1.3 или более поздняя версия требуется для экспорта 3D-геометрии с помощью wkb_hexсвойства.

import psycopg2
from shapely.geometry import LineString
from shapely import wkb

conn = psycopg2.connect('...')
curs = conn.cursor()

# Make a Shapely geometry
ls = LineString([(2.2, 4.4, 10.2), (3.3, 5.5, 8.4)])
ls.wkt  # LINESTRING Z (2.2 4.4 10.2, 3.3 5.5 8.4)
ls.wkb_hex  # 0102000080020000009A999999999901409A999999999911406666666666662440666666...

# Send it to PostGIS
curs.execute('CREATE TEMP TABLE my_lines(geom geometry, name text)')
curs.execute(
    'INSERT INTO my_lines(geom, name)'
    'VALUES (ST_SetSRID(%(geom)s::geometry, %(srid)s), %(name)s)',
    {'geom': ls.wkb_hex, 'srid': 4326, 'name': 'First Line'})

conn.commit()  # save data

# Fetch the data from PostGIS, reading hex-encoded WKB into a Shapely geometry
curs.execute('SELECT name, geom FROM my_lines')
for name, geom_wkb in curs:
    geom = wkb.loads(geom_wkb, hex=True)
    print('{0}: {1}'.format(name, geom.wkt))
# First Line: LINESTRING Z (2.2 4.4 10.2, 3.3 5.5 8.4)

Также отметим, что подобное может быть достигнуто путем отправки WKT геометрии, однако, поскольку она преобразуется в текст, это с потерями и может снизить точность точности. Передача геометрии в виде шестнадцатеричного кода WKB без потерь и сохраняет точную точность каждой координаты.

Майк Т
источник
Отлично, спасибо! Интересно, есть ли разница в производительности между этими двумя подходами.
Адам Матан
Существует 10% -ое преимущество в производительности с точками ST_MakePoint, что отлично подходит для геометрии точек. Тем не менее, создание более сложных типов геометрии, как правило, будет проще с Shapely.
Майк Т