Python deepcopy深拷贝时自定义GeoDataFrame子类类型丢失如何解决

自定义继承自GeoDataFrame的子类NodesGIS，WaterNetworkModel类的nodes_gis属性为NodesGIS类实例。对WaterNetworkModel实例执行deepcopy深拷贝后，副本的nodes_gis属性类型退化为GeoDataFrame，复现代码如下：

>>> type(wn.nodes_gis) # wn is an instance of WaterNetWorkModel
<class 'wntr.network.model.NodesGIS'>
>>> type(deepcopy(wn).nodes_gis)
<class 'geopandas.geodataframe.GeoDataFrame'>

geopandas的GeoDataFrame继承自pandas的DataFrame体系，这类容器类自定义了拷贝、切片、运算生成新实例的内部逻辑，默认逻辑不会自动感知用户自定义子类，会直接返回基类实例，最终导致子类类型丢失。

不需要修改WaterNetworkModel类的逻辑，直接在NodesGIS子类中实现pandas子类标准协议方法，从根源保证所有生成新实例的操作都保留子类类型即可：

1. 重写_constructor属性：这是pandas生态识别子类的核心配置，所有新建实例的操作都会调用该属性指向的构造器。
2. 重写__deepcopy__方法：明确深拷贝返回的实例类型，同时处理自定义属性的拷贝逻辑。

参考实现代码：

import geopandas as gpd
from copy import deepcopy

class NodesGIS(gpd.GeoDataFrame):
    # 若子类有自定义额外属性，可通过__slots__声明
    # __slots__ = ("custom_attr1", "custom_attr2")

    @property
    def _constructor(self):
        # 指定所有通用操作返回的实例类型为当前子类
        return NodesGIS

    def __deepcopy__(self, memo):
        # 调用父类深拷贝逻辑拿到基础GeoDataFrame副本
        base_copy = super().__deepcopy__(memo)
        # 构造子类实例，保留几何列、坐标系等核心元信息
        nodes_copy = NodesGIS(
            base_copy,
            geometry=self.geometry.name,
            crs=self.crs
        )
        # 子类有自定义实例属性的，在此处逐一复制到副本
        # nodes_copy.custom_attr1 = deepcopy(self.custom_attr1, memo)
        
        memo[id(self)] = nodes_copy
        return nodes_copy

实现上述逻辑后再执行深拷贝，即可正常保留NodesGIS类型：

class WaterNetworkModel:
    def __init__(self, nodes_gis):
        self.nodes_gis = nodes_gis

# 构造测试实例
test_nodes = NodesGIS(geometry=[], crs="EPSG:4326")
wn = WaterNetworkModel(nodes_gis=test_nodes)

print(type(wn.nodes_gis))  # 输出 <class '__main__.NodesGIS'>
print(type(deepcopy(wn).nodes_gis))  # 输出 <class '__main__.NodesGIS'>

• 不建议通过给WaterNetworkModel重写__deepcopy__手动强转nodes_gis类型的方式修复，这种方案无法覆盖单独拷贝NodesGIS实例的场景，后续维护成本更高。
• 如果NodesGIS重写了__init__方法、新增了自定义初始化参数，需要同步调整__deepcopy__里的构造传参，保证副本初始化逻辑和原对象完全一致。

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