如何使用带有 *args 的映射在 python 函数调用中解包元组

Question

我目前正在对一组我想要并行化的变量进行合并。我的代码如下所示：

mergelist = [
  ('leftfile1', 'rightfile1', 'leftvarname1', 'outputname1'),
  ('leftfile1', 'rightfile1', 'leftvarname2', 'outputname2')
  ('leftfile2', 'rightfile2', 'leftvarname3', 'outputname3')
]

def merger(leftfile,rightfile,leftvarname,outvarname):
   do_the_merge

for m in mergelist:
     merger(*m)

通常，为了加快长循环，我会将 for m in mergelist 替换为...。

from multiprocessing import Pool

p = Pool(8)
p.map(merger(m), mergelist)
p.close()

但由于我使用星号来解包元组，我不清楚如何正确映射它。如何获得*m？

Answer 1

使用lambda:

with Pool(8) as p:
    p.map(lambda m:merger(*m), mergelist)

Answer 2

恕我直言，最简单的解决方案是更改合并功能，或添加包装器：

def merger(leftfile,rightfile,'leftvarname','outvarname'):
    do_the_merge

def merger_wrapper(wrapper_tuple):
    merger(*wrapper_tuple)

p.map(merger_wrapper, mergelist)

我看到@delnan 实际上也将这个解决方案放在了 cmets 中。

为了增加一点价值：）你也可以这样包装它：

from functools import partial
def unpack_wrapper(f):
    def unpack(arg):
        return f(*arg)
    return unpack

这应该可以让您将其简化为

p.map(unpack_wrapper(merger), mergelist)

Answer 3

您可以在合并函数中解压元组：

def merger(args):
    if len(args) != 4:
         # error
    leftfile,rightfile,leftvarname,outvarname = args
    do_the_merge

另一种选择是在参数列表中解包：

def merger( (leftfile,rightfile,leftvarname,outvarname) ):
    do_the_merge

编辑：解决 OP 问题：

def merger((l,r,v,o)):
    return l+r

for m in mergelist:
         print merger(m)

返回

leftfile1rightfile1
leftfile1rightfile1
leftfile2rightfile2