使用gdf.set_geometry()无法指定GeoDataFrame几何列的问题排查

你用嵌套列表直接创建GeoDataFrame时，GeoPandas不会自动把含shapely几何对象的列标记为几何数据类型，这些列只会被当成普通Python对象列。所以调用set_geometry时，GeoPandas找不到符合要求的几何列，就会报错。

方法1：创建时显式指定几何列（推荐） #

创建GeoDataFrame时用字典格式传数据，同时通过geometry参数指定初始活跃几何列，这样所有含几何对象的列都会被正确识别：

from geopandas import GeoDataFrame as gdf
from shapely.geometry import Polygon, Point

square = Polygon([[0,0], [0,1], [1,1], [1,0]])
triangle = Polygon([[2,0], [4,0], [3,1]])

data = {
    "shape": [square, triangle],
    "name": ["kwadrat", "trójkąt"],
    "center": [Point([0.5,0.5]), Point([3,0.3])]
}

# 创建时指定初始几何列，其他几何列自动被识别
figury = gdf(data, geometry="shape")

# 自由切换活跃几何列，记得重新赋值
figury = figury.set_geometry("center")

方法2：先转换列类型再设置 #

如果已经创建了GeoDataFrame，先把目标列转成几何类型，再调用set_geometry：

# 将普通列转换为几何类型
figury["shape"] = gdf(figury["shape"]).geometry
figury["center"] = gdf(figury["center"]).geometry

# 现在可以设置活跃几何列
figury = figury.set_geometry("shape")
# 切换到center列也没问题
figury = figury.set_geometry("center")

重要提醒 #

• set_geometry默认返回新的GeoDataFrame，不会修改原对象，必须把结果重新赋值给变量，否则原数据不会有变化。
• 只有列类型是geopandas.geometry时，set_geometry才能识别并设为活跃列。

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