如何使用joblib Python返回矩陣？

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) 

我想val存儲所有retruned列出了每個迭代從1到500，然後使用它們。

沒有平行，我會做它：

def myFunc(c, K, N): 
    # something 
    return myList 
inputs = range(500) 
lists = [] 
for c in inputs: 
    val = myFunc(c, K, N) 
    lists.append(val) 
# Now I have all the lists 

我能做到這一點與並行？

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) 

這將打印

[[0, 1, 1, 0, 1, 2, 2, 1, 1], [2, 1, 2, 0, 1, 1, 2, 1, 0]]