如何使用joblibpython返回矩阵？

Question

我有一个返回列表的函数。我使用joblib进行并行处理。在

from joblib import Parallel, delayed
def myFunc(c, K, N):
    # something
    return myList
inputs = range(500)
if __name__ == '__main__':
    val = Parallel(n_jobs=20)(delayed(myFunc)(c, K, N) for c in inputs)

我想在从1到500的每个迭代中存储所有重新运行的列表val，然后使用它们。在

如果没有平行线，我会这样做：

^{pr2}$

我能用平行线吗？在

Answer 1

事实上，joblib.Parallel为你做这项工作。在

from __future__ import print_function
from joblib import Parallel, delayed
import random


def my_func(c, K, N):
    my_list = [random.randrange(3) for _ in range(K*N)]
    return my_list


if __name__ == '__main__':
    inputs = range(2)
    K, N = 3, 3
    val = Parallel(n_jobs=2)(delayed(my_func)(c, K, N) for c in inputs)
    print(val)

这会打印出来

^{pr2}$