如何从深层复制中排除特定引用？

Question

我有一个对象，它有自己的内容（即某物的列表）和对另一个对象的引用，它与其链接。如何排除对另一个对象的引用进行深度复制？

from copy import deepcopy
class Foo:
    def __init__(self, content, linked_to):
        self.content = content
        self.linked_to = linked_to

a1 = Foo([[1,2],[3,4]], None)
a2 = Foo([[5,6],[7,8]], a1)

a3 = deepcopy(a2) # <- I don't want there, that a3.linked_to will be copied
# I want that a3.linked_to will still point to a1

a3.linked_to.content.append([9,10])
print a1.content # [[1,2],[3,4]], but I want [[1,2],[3,4], [9,10]] 

Answer 1

您的类可以实现一个__deepcopy__方法来控制它的复制方式。来自copy模块文档：

为了让类定义自己的复制实现，它可以定义特殊的方法__copy__()和__deepcopy__(). 前者被调用来实现浅拷贝操作；没有传递额外的参数。调用后者来实现深拷贝操作；它传递一个参数，即备忘录字典。如果__deepcopy__()实现需要对组件进行深层复制，则应调用该deepcopy()函数，并将该组件作为第一个参数，将备忘录字典作为第二个参数。

只需返回您的类的一个新实例，以及您不想进行深度复制的引用，只需按原样获取即可。使用该deepcopy()函数复制其他对象：

from copy import deepcopy

class Foo:
    def __init__(self, content, linked_to):
        self.content = content
        self.linked_to = linked_to

    def __deepcopy__(self, memo):
        # create a copy with self.linked_to *not copied*, just referenced.
        return Foo(deepcopy(self.content, memo), self.linked_to)

演示：

>>> a1 = Foo([[1, 2], [3, 4]], None)
>>> a2 = Foo([[5, 6], [7, 8]], a1)
>>> a3 = deepcopy(a2)
>>> a3.linked_to.content.append([9, 10])  # still linked to a1
>>> a1.content
[[1, 2], [3, 4], [9, 10]]
>>> a1 is a3.linked_to
True
>>> a2.content is a3.content  # content is no longer shared
False