pickle - 不能腌制 instancemethod 对象

Question

我遇到了泡菜的问题，代码是：

import cPickle

class A(object):

    def __init__(self):
        self.a = 1

    def methoda(self):
        print(self.a)


class B(object):

    def __init__(self):
        self.b = 2
        a = A()
        self.b_a = a.methoda

    def methodb(self):
        print(self.b)
if __name__ == '__main__':
    b = B()
    with open('best_model1.pkl', 'w') as f:
        cPickle.dump(b, f)

错误是：

文件“/usr/lib/python2.7/copy_reg.py”，第 70 行，在 _reduce_ex 中引发 TypeError，“无法腌制 %s 对象”% base。名称类型错误：无法腌制实例方法对象

Answer 1

如果您使用dill而不是cPickle.

>>> import dill     
>>> 
>>> class A(object):
...   def __init__(self):
...     self.a = 1
...   def methods(self):
...     print(self.a)
... 
>>> 
>>> class B(object):
...   def __init__(self):
...     self.b = 2
...     a = A()
...     self.b_a = a.methods
...   def methodb(self):
...     print(self.b)
... 
>>> b = B()
>>> b_ = dill.dumps(b)
>>> _b = dill.loads(b_)
>>> _b.methodb()
2
>>> 

另请参阅： Can't pickle <type 'instancemethod'> when using python's multiprocessing Pool.map()

Answer 2

此外，当安装 dill 时，pickle 会起作用，但 cPickle 不会像往常一样。

import cPickle, pickle

class A(object):
    def __init__(self):
        self.a = 1
    def methoda(self):
        print(self.a)

class B(object):
    def __init__(self):
        self.b = 2
        a = A()
        self.b_a = a.methoda
    def methodb(self):
        print(self.b)

# try using cPickle
try:
    c = cPickle.dumps(b)
    d = cPickle.loads(c)
except Exception as err:
    print('Unable to use cPickle (%s)'%err)
else:
    print('Using cPickle was successful')
    print(b)
    print(d)

# try using pickle
try:
    c = pickle.dumps(b)
    d = pickle.loads(c)
except Exception as err:
    print('Unable to use pickle (%s)'%err)
else:
    print('Using pickle was successful')
    print(b)
    print(d)

 >>> Unable to use cPickle (can't pickle instancemethod objects)
 >>> Using pickle was successful
 >>> <__main__.B object at 0x10e9b84d0>
 >>> <__main__.B object at 0x13df07190>

无论出于何种原因，cPickle 不仅仅是一个 C 版本的 pickle 快 100 倍，但有一些差异