写入带多层索引列的GeoDataFrame时触发TypeError的排查与解决

问题重现 #

创建GeoDataFrame的代码如下：

import pandas as pd
import geopandas
df = pd.DataFrame([[1, 0], [0, 1]], columns=[['location', 'location'], ['x', 'y']])
df["geometry"] = geopandas.points_from_xy(df['location','x'], df['location', 'y'])
gdf = geopandas.GeoDataFrame(df, geometry="geometry", crs="EPSG:4326")

调用gdf.to_file("test.shp")时触发错误：

TypeError: Cannot interpret '<geopandas.array.GeometryDtype object at 0x7f99fdf45070>' as a data type

错误原因 #

核心问题是GeoDataFrame使用了多层列索引（MultiIndex），而Shapefile格式本身不支持多层结构的列名。当to_file方法尝试处理这种列结构时，会因为无法解析多层列对应的数据类型而抛出类型错误。

解决方法 #

只需将多层列索引转换为单层列名即可，有两种常见方式：

方式1：扁平化多层列名 #

通过拼接多层列的名称生成单层列名：

# 自动拼接多层列名，用下划线分隔
gdf.columns = ['_'.join(col) if isinstance(col, tuple) else col for col in gdf.columns]
# 执行后列名变为 ['location_x', 'location_y', 'geometry']

方式2：手动指定新列名 #

直接为每一列设置明确的单层名称：

gdf.columns = ['location_x', 'location_y', 'geometry']

修改完成后，再次调用gdf.to_file("test.shp")即可正常导出Shapefile文件。

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