Python使用多个输入对函数进行多处理

Question

在Python中，该multiprocessing模块可用于在一系列值上并行运行函数。例如，这将生成f的前100000个评估的列表。

def f(i):
    return i * i

def main():
    import multiprocessing
    pool = multiprocessing.Pool(2)
    ans = pool.map(f, range(100000))

    return ans

当f接受多个输入而只有一个变量变化时，是否可以做类似的事情？例如，如何并行处理此：

def f(i, n):
    return i * i + 2*n

def main():
    ans = []
    for i in range(100000):
        ans.append(f(i, 20))

    return ans

Answer 1

有几种方法可以做到这一点。在问题给出的示例中，您可以定义一个包装函数

def g(i):
    return f(i, 20)

然后将此包装传递给map()。一种更通用的方法是使用一个包装器，该包装器使用一个元组参数并将该元组解包为多个参数

def g(tup):
    return f(*tup)

或使用等效的lambda表达式：lambda tup: f(*tup)。