不同版本Geopandas间Pickle序列化反序列化兼容性问题

问题背景 #

用geopandas 0.7.0序列化的GeoDataFrame，在geopandas 0.10.2环境中读取时触发KeyError: 0。根源是geopandas 0.8版本开始引入pygeos作为底层几何处理库，导致GeoSeries的序列化结构发生变更：

• 0.7.0及更早版本的GeoSeries序列化后，state是包含几何数据和CRS的字典结构
• 0.10.2版本（启用pygeos时）期望state是元组格式，尝试按索引0取值时找不到对应键，因此抛出KeyError

解决方案 #

方法1：低版本环境读取后转存兼容格式（最稳妥） #

Pickle本身对版本依赖极强，转存为GeoJSON或Parquet这类跨版本友好的格式是长期解决方案：

1. 搭建与序列化时一致的环境（geopandas 0.7.0、pandas 1.0.3等）
2. 读取原始Pickle文件：

import geopandas as gpd
gdf = gpd.read_pickle("geo_data_frame.pkl")

3. 保存为兼容格式：

# 保存为GeoJSON（无需额外依赖）
gdf.to_file("geo_data_frame.geojson", driver="GeoJSON")

# 或保存为Parquet（需安装pyarrow/fastparquet）
gdf.to_parquet("geo_data_frame.parquet")

4. 在目标环境（geopandas 0.10.2）中读取新文件：

# 读取GeoJSON
gdf = gpd.read_file("geo_data_frame.geojson")

# 读取Parquet
gdf = gpd.read_parquet("geo_data_frame.parquet")

方法2：高版本环境修补反序列化逻辑（临时方案） #

如果无法搭建低版本环境，可临时修改GeoArray的反序列化方法兼容旧格式：

import pickle
import geopandas as gpd
from geopandas.array import GeoArray

# 备份原始方法
original_setstate = GeoArray.__setstate__

def patched_setstate(self, state):
    # 识别旧版本的字典格式state，转为新版本期望的元组格式
    if isinstance(state, dict):
        wkb_data = state.get('data')
        crs = state.get('crs')
        state = (wkb_data, crs)
    original_setstate(self, state)

# 替换反序列化方法
GeoArray.__setstate__ = patched_setstate

# 读取Pickle文件
gdf = gpd.read_pickle("geo_data_frame.pkl")

# 可选：恢复原始方法
GeoArray.__setstate__ = original_setstate

注意：该方法依赖geopandas内部实现，后续版本可能失效，仅作临时救急。

方法3：禁用pygeos强制使用旧逻辑 #

在高版本环境中通过环境变量禁用pygeos，强制使用旧版shapely/fiona的处理路径：

import os
os.environ['USE_PYGEOS'] = '0'

import geopandas as gpd
gdf = gpd.read_pickle("geo_data_frame.pkl")

注意：geopandas 0.10.x之后逐步弃用非pygeos路径，后续版本可能不再支持此方法。

内容的提问来源于stack exchange，提问作者Rock Dev Mufasa