Wstaw punkt do PostGIS za pomocą Pythona

11

Jaki jest właściwy sposób wstawienia punktu do bazy danych PostGIS przy użyciu Pythona?

Adam Matan
źródło

Odpowiedzi:

21

Najpierw zainstaluj pakiet psycopg2 , interfejs Pythonic dla PostgreSQL.

Następnie użyj 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 można użyć do sprawdzenia poprawności wartości:

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

Notatki

  • Pamiętajcie, że (lat, lon)tak (y, x)nie jest (x, y).
  • Zawsze używaj parametrów zamiast manipulacji ciągiem, aby zapobiec wstrzykiwaniu SQL . W tych przykładach mamy tup (x, y, z)na końcu, aby psycopg2obsłużyć podstawienie.
Adam Matan
źródło
20

W przypadku bardziej skomplikowanych geometrii, takich jak geometria LineString i Wielokąt, możesz obsługiwać je za pomocą Shapely, a następnie przepuszczać je przez psycopg2 jako WKB z kodowaniem heksadecymalnym. Pamiętaj, że Shapely 1.3 lub nowszy jest wymagany do obsługi eksportu geometrii 3D z tą wkb_hexwłaściwością.

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)

Ponadto zauważ, że podobne można osiągnąć, wysyłając WKT geometrii, jednak ponieważ jest on konwertowany na tekst, jest stratny i może zmniejszać angstremę precyzji. Przenoszenie geometrii jako WKB z kodowaniem heksadecymalnym jest bezstratne i zachowuje dokładność każdej współrzędnej.

Mike T.
źródło
Wspaniale, dziękuję! Zastanawiam się, czy istnieje różnica w wydajności między tymi dwoma podejściami.
Adam Matan,
Z 10-punktowymi punktami przewagi wydajnościowej ST_MakePoint, co jest świetne dla geometrii punktów. Jednak budowanie bardziej złożonych typów geometrii będzie ogólnie łatwiejsze dzięki Shapely.
Mike T