如何让 Pool.map 接受一个 lambda 函数

Question

我有以下功能：

 def copy_file(source_file, target_dir):
    pass

现在我想使用 multiprocessing 立即执行这个函数：

 p = Pool(12)
p.map(lambda x: copy_file(x,target_dir), file_list)

问题是，lambda 不能被腌制，所以失败了。解决此问题的最简洁（pythonic）方法是什么？

原文由 Peter Smit 发布，翻译遵循 CC BY-SA 4.0 许可协议

Answer 1

使用函数对象：

 class Copier(object):
    def __init__(self, tgtdir):
        self.target_dir = tgtdir
    def __call__(self, src):
        copy_file(src, self.target_dir)

运行你的 Pool.map ：

 p.map(Copier(target_dir), file_list)

原文由 Fred Foo 发布，翻译遵循 CC BY-SA 3.0 许可协议

Answer 2

对于 Python2.7+ 或 Python3，您可以使用 functools.partial ：

 import functools
copier = functools.partial(copy_file, target_dir=target_dir)
p.map(copier, file_list)

原文由 unutbu 发布，翻译遵循 CC BY-SA 4.0 许可协议