如何高效利用PostGIS的IN子句匹配GeoDataFrame几何对象

需要找出GeoDataFrame中的哪些几何对象已存在于PostGIS数据表中，原本计划用SQL的IN子句实现，但不清楚如何针对PostGIS几何对象使用该子句。目前已通过将GeoDataFrame转换为WKB格式，并对每个几何对象调用ST_GeomFromWKB()实现需求，SQL语句如下：

select id, geom from table
where geom in (ST_GeomFromWKB(geom1), ST_GeomFromWKB(geom2), ...)

但这种方式效率较低，希望找到更优方案（比如利用ST_DWithin()），并实现类似如下简洁的SQL语句：

select id, geom from table
where geom in (geom1, geom2, geom3)

1. 临时表JOIN法（推荐，大数量场景高效） #

将GeoDataFrame的几何对象批量导入PostGIS临时表，再通过JOIN匹配，能充分利用空间索引提升效率：

• 第一步：用GeoPandas将GeoDataFrame导入临时表

gdf.to_postgis('temp_geoms', your_db_connection, if_exists='replace', index=False)

• 第二步：执行匹配查询（用ST_Equals保证几何严格相等，或直接用=）

SELECT t.id, t.geom
FROM your_target_table t
JOIN temp_geoms tmp ON ST_Equals(t.geom, tmp.geom);

2. 批量参数化查询（简洁高效） #

利用PostgreSQL的ANY操作符配合参数化传递几何数组，比逐个调用ST_GeomFromWKB更高效：

import psycopg2
from shapely.wkb import dumps

# 提取GeoDataFrame中几何的WKB字节串
geom_wkbs = [dumps(geom) for geom in gdf.geometry]

# 执行参数化查询
conn = psycopg2.connect(your_db_params)
cur = conn.cursor()
query = """
SELECT id, geom
FROM your_target_table
WHERE geom = ANY(%s);
"""
cur.execute(query, (geom_wkbs,))
results = cur.fetchall()

3. 近似匹配用ST_DWithin（适用于有坐标误差的场景） #

如果几何对象存在微小坐标偏差，无法严格匹配，可使用ST_DWithin设置距离阈值（设为0等价于严格匹配，但能利用空间索引）：

SELECT t.id, t.geom
FROM your_target_table t
JOIN temp_geoms tmp ON ST_DWithin(t.geom, tmp.geom, 0);

前提：目标表和临时表的geom列都已创建空间索引

内容的提问来源于stack exchange，提问作者Lucas Fujita