解压dictionary dictionary参数，并在多处理池中使用**kwargs

Question

all！

通常，如果我需要将字典作为参数集传递给函数，一种简单的方法是打开字典并使用它们。

def some_function(a=0, b=0, **params):
print(a) # 1
print(b) # 2
d = {"a":1, "b":2}
some_function(**d)

我正在做一个需要很长时间的小项目。因此，我试图使用Multiprocessing来帮助我更快地完成工作。我简化了代码(如下(。

from multiprocessing import Pool
from itertools import repeat
import time
def some_function(n, param1=0, param2=0):
#some operation on n and **params
print(n)
print(param1)
input_k = [i for i in range(1,10,2)]
input_params = {'param1':1, 'param2':1}
if __name__ == '__main__':
with Pool(16) as p:
k = p.starmap(some_function, zip(input_k, repeat(input_params)))
############## output ##############
#1
#7
#9
#5
#3
#{'param1': 1, 'param2': 1}
#{'param1': 1, 'param2': 1}
#{'param1': 1, 'param2': 1}
#{'param1': 1, 'param2': 1}
#{'param1': 1, 'param2': 1}

我想在相应的input_k和input_params上运行那个some_function。但是函数的输出并不像预期的那样。看起来函数取了整个input_params字典，并将其分配给函数中的param1？

我知道我在将字典传递给函数时没有对其进行解包，但我不知道如何做到这一点(zip()不允许我简单地将"**"添加到the input_param(。我该怎么做？

谢谢！

Answer 1

我将您的代码修改为：

from multiprocessing import Pool
from itertools import repeat
import time
def some_function(n, param1=0, param2=0):
#some operation on n and **params
print(n, param1, param2)
input_k = [i for i in range(1,10,2)]
input_params = {'param1':1, 'param2':1}
if __name__ == '__main__':
with Pool(16) as p:
k = p.starmap(some_function,  [[x,*y.values()] for x,y in zip(input_k, repeat(input_params))])

现在运行良好。