python - 在 Python 中实现并行 for 循环

Question

我有一个如下所示的 Python 程序:

total_error = []
for i in range(24):
    error = some_function_call(parameters1, parameters2)
    total_error += error

函数“some_function_call”需要很多时间，我找不到降低函数时间复杂度的简单方法。有没有办法在执行并行任务时仍然减少执行时间，然后将它们加到 total_error 中。我尝试使用 pool 和 joblib，但都无法成功使用。

Answer 1

您还可以使用 concurrent.futures在 Python 3 中，这是一个比 multiprocessing 更简单的接口(interface). See this有关差异的更多详细信息。

from concurrent import futures

total_error = 0

with futures.ProcessPoolExecutor() as pool:
  for error in pool.map(some_function_call, parameters1, parameters2):
    total_error += error

在这种情况下，parameters1和 parameters2应该是一个列表或可迭代的，其大小与您要运行该函数的次数相同(根据您的示例为 24 次)。

如果paramters<1,2>不是可迭代/可映射的，但您只想运行该函数 24 次，您可以提交该函数的作业达到所需的次数，然后使用回调获取结果。

class TotalError:
    def __init__(self):
        self.value = 0

    def __call__(self, r):
        self.value += r.result()

total_error = TotalError()
with futures.ProcessPoolExecutor() as pool:
  for i in range(24):
    future_result = pool.submit(some_function_call, parameters1, parameters2)
    future_result.add_done_callback(total_error)

print(total_error.value)