如何使用GeoPandas创建、保存并加载空间索引以避免重复构建？

GeoPandas本身没有直接提供保存空间索引的API，但可以借助Python的pickle模块序列化索引对象，实现一次创建多次复用。以下是具体步骤：

1. 创建并保存空间索引 #

先加载你的GeoDataFrame，生成空间索引后用pickle写入文件：

import geopandas as gpd
import pickle

# 加载目标GeoDataFrame
gdf = gpd.read_file("your_data.shp")

# 生成空间索引（访问sindex属性时会自动创建）
spatial_index = gdf.sindex

# 将索引保存到本地文件
with open("spatial_index.pkl", "wb") as f:
    pickle.dump(spatial_index, f)

2. 加载已保存的索引并复用 #

后续使用时，需加载与创建索引时完全一致的GeoDataFrame（几何列不能有修改），再加载索引并赋值给GeoDataFrame的sindex属性：

import geopandas as gpd
import pickle

# 加载同一份GeoDataFrame
gdf = gpd.read_file("your_data.shp")

# 读取保存的空间索引
with open("spatial_index.pkl", "rb") as f:
    saved_index = pickle.load(f)

# 将加载的索引赋值给GeoDataFrame的sindex属性
gdf.sindex = saved_index

# 直接使用索引进行空间查询，示例：
matches_idx = list(gdf.sindex.query(target_geometry, predicate="intersects"))
matches = gdf.iloc[matches_idx]

关键注意事项 #

• 若GeoDataFrame的几何数据（如新增/删除要素、修改几何形状）发生变化，之前保存的索引会直接失效，必须重新创建并保存。
• 保存和加载索引时，尽量保持Python、GeoPandas、rtree的版本一致，避免序列化兼容性问题。

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