仅在一个参数上并行化多参数 python 代码

Question

我有一个需要 7 个参数的 python 函数：

my_fun(a,b,c,d,e,f,g)

我想并行运行它，但保持从 b 到 g 的参数固定并迭代 a。

目前我找到的解决方案是将其作为子函数运行：

def parallel_funct(t1):
    a = t1 
    b = list
    c = string
    d = list
    e= list
    f = list
    g = string
    stationxml_storage = string
    my_fun(a,b,c,d,e,f,g)

然后我运行：

ts = np array

if __name__ == '__main__':
    p = Pool(6)
    p.map(parallel_funct, ts)

有没有办法只使用 my_fun 来做到这一点？

希望很明确

Answer 1

稍微更优雅的方法是使用functools.partial 函数，避免仅仅为了部分参数应用而定义顶级函数；但是没有办法直接告诉Pool.map 你想要并行化哪个参数。