如何将FeatureCollection插入PostGIS？SQLAlchemy实操报错

问题根源 #

你代码里的错误主要是两个点：

1. 变量混乱：用到了gs、sat这类未定义的变量，和实际的gdf、Calculation实例不匹配
2. 类型错误：直接把整个GeoDataFrame对象传给shape字段，PostGIS需要的是单个几何对象对应的GeoJSON字符串，没法直接适配GeoDataFrame类型

方法1：逐行遍历插入 #

先把GeoJSON正确解析成GeoDataFrame，然后循环每一行，提取几何对象的GeoJSON字符串，用ST_GeomFromGeoJSON转换后插入数据库：

import geopandas as gpd
import json
from sqlalchemy import func
# 替换成你自己的模块导入路径
from your_model_module import Calculation
from your_db_module import db

# 加载FeatureCollection数据
js = '''
{
  "type": "FeatureCollection",
  "features": [
    // 你的要素内容
  ]
}
'''
geojson_data = json.loads(js)
gdf = gpd.GeoDataFrame.from_features(geojson_data["features"])

# 遍历每一行创建实例并插入
for _, row in gdf.iterrows():
    # 把几何对象转成GeoJSON字符串
    geom_json = row.geometry.to_json()
    calc = Calculation(
        tle='1',  # 可根据实际需求修改，比如从row的属性中提取对应值
        shape=func.ST_GeomFromGeoJSON(geom_json)
    )
    db.add(calc)

# 提交事务
db.commit()

方法2：批量插入（大数据量推荐） #

如果数据量较大，用GeoPandas的to_sql批量插入效率更高，需要确保环境安装了geoalchemy2来处理空间类型适配：

import geopandas as gpd
import json
from sqlalchemy import create_engine
from geoalchemy2 import Geometry

# 替换成你的数据库连接字符串
engine = create_engine("postgresql://username:password@host:port/dbname")

# 加载GeoDataFrame
js = '''{ /* 你的FeatureCollection内容 */ }'''
geojson_data = json.loads(js)
gdf = gpd.GeoDataFrame.from_features(geojson_data["features"])

# 添加tle列，这里统一设为'1'，可根据实际逻辑调整
gdf['tle'] = '1'

# 批量写入数据库
gdf.to_sql(
    name='calculations',
    con=engine,
    if_exists='append',  # 可选值：fail（表存在则报错）、replace（覆盖表）、append（追加数据）
    dtype={
        'shape': Geometry(geometry_type='GEOMETRY', srid=4326)
    },
    index=False  # 不将GeoDataFrame的索引写入数据库
)

额外注意事项 #

• 必须安装依赖：pip install geopandas geoalchemy2 psycopg2-binary
• 如果你的GeoJSON中是单一几何类型（比如全是多边形），可以把geometry_type改成POLYGON，类型更精准
• 批量插入时注意if_exists参数的选择，避免误删数据或重复插入

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