使用 cython 的 deepcopy 函数的问题 类

issue using deepcopy function for cython classes

我最近一直在使用 Cython 来提高速度,但是当我尝试使用 copy.deepcopy() 时出现一些错误 occurred.Here 是代码:

from copy import deepcopy

cdef class cy_child:
    cdef public:
        int move[2]
        int Q
        int N
    def __init__(self, move):
        self.move = move
        self.Q = 0
        self.N = 0      

a = cy_child((1,2))

b = deepcopy(a)

这是错误:
can't pickle _cython_magic_001970156a2636e3189b2b84ebe80443.cy_child objects

如何解决此代码的问题?

正如 hpaulj 在评论中所说,deepcopy 看起来默认使用 pickle 来完成它的工作。 Cython cdef classes 过去不是可腌制的。 In recent versions of Cython they are where possible (see also http://blog.behnel.de/posts/whats-new-in-cython-026.html) 但是酸洗数组看起来是个问题(即使没有数组我也无法正常工作)。

解决办法是自己实现相关功能。我已经完成 __deepcopy__ since it's simple but alternatively you could implement the pickle protocol

def __deepcopy__(self,memo_dictionary):
    res = cy_child(self.move)
    res.Q = self.Q
    res.N = self.N
    return res

我怀疑您将来不需要这样做,因为 Cython 改进了他们的 pickle 实现。


关于 memo_dictionary 的注释:假设您有

a=[None]
b=[A]
a[0]=B
# i.e. A contains a link to B and B contains a link to A
c = deepcopy(a)

memo_dictionarydeepcopy 用来记录它已经复制的内容,这样它就不会永远循环。您不需要自己做很多事情。但是,如果您的 cdef class 包含一个 Python 对象(包括另一个 cdef class),您应该像这样复制它:

cdef class C:
    cdef object o
    def __deepcopy__(self,memo_dictionary):
        # ...
        res.o = deepcopy(self.o,memo_dictionary)
        # ...

(即确保它被传递给 deepcopy 的进一步调用)