python - 自定义dict的深拷贝deepcopy()

最新推荐文章于 2025-03-21 23:31:32 发布

小么额菇

最新推荐文章于 2025-03-21 23:31:32 发布

阅读量3.1k

点赞数 2

CC 4.0 BY-SA版权

分类专栏：学习笔记文章标签： python deepcopy

本文链接：https://blog.csdn.net/Joyce__Yin/article/details/108123556

学习笔记专栏收录该内容

27 篇文章

订阅专栏

本文详细解析了当自定义的class继承dict时，如何正确实现__deepcopy__方法以避免浅拷贝或死循环的问题。通过具体示例，讲解了利用memo字典跟踪拷贝对象，确保深拷贝的正确性和效率。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

当有自定义的class继承了dict时，在使用copy.deepcopy便需要重写__deepcopy__方法，这个方法一不小心就会写成shallow copy(只copy了最外层) 或是写出死循环，如下反例：

    def __deepcopy__(self, memo=None, _nil=[]):
        dict = _MyDict()
        for key in self.keys():
            dict.__setattr__(key,self.__getattr__(key))

注意到参数中实际上有一个memodict，是可以用来避免死循环的，字典copy的主要过程首先是这样：

for key in self.keys():
    dict .__setattr__(deepcopy(key, memo),
                      deepcopy(self.__getattr__(key), memo))

key和value都需要调用各自的deepcopy方法，之后在函数入口处应当首先判断当前对象是否在memo中出现过（这里写法参照deepcopy函数源码），即，是否是递归的循环调用，如果没有，需要将自己的id加到字典中，表明自己正在进行deep copy：

        if memo is None:
            memo = {}
        d = id(self)
        y = memo.get(d, _nil)
        if y is not _nil:
            return y

        dict = _MyDict()
        memo[d] = id(dict)

整体代码如下：

class _MyDict(dict):
    ....

    def __deepcopy__(self, memo=None, _nil=[]):
        if memo is None:
            memo = {}
        d = id(self)
        y = memo.get(d, _nil)
        if y is not _nil:
            return y

        dict = _MyDict()
        memo[d] = id(dict)
        for key in self.keys():
            dict .__setattr__(deepcopy(key, memo),
                              deepcopy(self.__getattr__(key), memo))
        return dict