Question

我最近一直在玩Cython以加快速度，但当我尝试使用copy.deepcopy()时发生了一些错误。这是代码：

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

如何解决此代码的问题？

Answer 1

正如hpaulj在评论中所述，deepcopy默认使用pickle来完成其工作。 Cython cdef class es曾经不是可以选择的。 In recent versions of Cython they are where possible（另请参阅http://blog.behnel.de/posts/whats-new-in-cython-026.html）但是挑选数组似乎是一个问题（即使没有数组，我也没有让它工作）。

解决方案是自己实现相关功能。我已经完成__deepcopy__因为它很简单，但您可以实现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_dictionary使用

deepcopy来记录已经复制的内容，以便它不会永远循环。你自己不需要做太多的事情。但是，如果您的cdef类包含一个Python对象（包括另一个cdef类），您应该像这样复制它：

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

（即确保将其传递给deepcopy）

的进一步调用

使用deepcopy函数进行cython类的问题

1 个答案: