Question

是否有一个函数或类能够将lambda函数作为参数并返回可以使用Pickle进行pickle的对象或函数？

我正在尝试使用joblib.Parallel来并行化一些代码，这些代码有一个lambda函数的问题，并且将所有lambda函数转换为命名函数会使我的代码看起来比以前更糟糕。

Answer 1

请注意，腌制函数只会腌制其名称以及模块的名称：

In [16]: import pickle

In [19]: def foo(x): return x+1

In [20]: pickle.dumps(foo)
Out[20]: 'c__main__\nfoo\np0\n.'

所以要腌制一个函数，你需要给它一个名字。取消功能需要导入模块。

如果命名所有lambda函数“使代码看起来更糟”，那么可能会将它们隐藏在一个单独的模块中，或者考虑如何推广这些函数以减少它们的数量。

Answer 2

您可以使用dill代替pickle，它知道如何序列化lambda，即使在函数或类中用作默认参数时也是如此。

>>> import dill
>>> 
>>> def doit(x, y=lambda x:x+1):
...   return y(x)
... 
>>> doit(3)
4
>>> _ = dill.dumps(doit)
>>> _
'\x80\x02cdill.dill\n_create_function\nq\x00(cdill.dill\n_load_type\nq\x01U\x08CodeTypeq\x02\x85q\x03Rq\x04(K\x02K\x02K\x02KCU\n|\x01\x00|\x00\x00\x83\x01\x00Sq\x05N\x85q\x06)U\x01xq\x07U\x01yq\x08\x86q\tU\x07<stdin>q\nU\x04doitq\x0bK\x01U\x02\x00\x01q\x0c))tq\rRq\x0ec__builtin__\n__main__\nh\x0bh\x00(h\x04(K\x01K\x01K\x02KCU\x08|\x00\x00d\x01\x00\x17Sq\x0fNK\x01\x86q\x10)h\x07\x85q\x11U\x07<stdin>q\x12U\x08<lambda>q\x13K\x01U\x00q\x14))tq\x15Rq\x16c__builtin__\n__main__\nh\x13NN}q\x17tq\x18Rq\x19\x85q\x1aN}q\x1btq\x1cRq\x1d.'
>>> didit = dill.loads(_)
>>> didit(3)
4
>>>

请注意，嵌套的lambdas和闭包也可以正常工作：

>>> f = lambda x: lambda y: x+y
>>> f(1)(2)
3
>>> dill.loads(dill.dumps(f))(1)(2)
3
>>> dill.loads(dill.dumps(f(1)))(2)
3
>>>

我知道至少有joblib的分支使用dill代替pickle，这可能就是为什么你首先要问pickle。如果找不到分支，请查找cloudpickle的分支，该dill类似于use [DB_1] CREATE FUNCTION ADD_FIVE(@id varchar(10)) RETURNS int BEGIN DECLARE @new_value int; SELECT @new_value = 5+value FROM my_table WHERE ID = @id; RETURN @new_value; END GO --now synonym use [DB_2] CREATE SYNONYM dbo.ADD_FIVE FOR DB_1.dbo.ADD_FIVE GO，这也可能有用。

将lambda函数转换为pickleable对象

2 个答案: