python并行执行中变量的范围

Question

我的问题用一个简单的代码总结如下：

#!/usr/bin/env python
from joblib import Parallel, delayed
import numpy as np
h =  np.ones(3)
def func(i):
    global h
    h[i] = h[i]+i
    print(h[i])
print(h)
Parallel(n_jobs=3)(delayed(func)(i) for i in range(3))
print(h)

输出为：

[ 1.  1.  1.]
1.0
2.0
3.0
[ 1.  1.  1.]

但是，我希望在循环内修改h值时对它们进行修改。 我究竟做错了什么？

编辑：我尝试了L_S建议的答案backend="threading" 。 它达到了目的。 但是，代码以串行方式运行。 循环从3个n_jobs开始，我只能看到一个python运行在top ，而有3个python可执行文件在未指定backend="threading"情况下运行。

Answer 1

默认情况下，Parallel使用Python多处理模块来分叉单独的Python工作进程，以在单独的CPU上同时执行任务。

因此，您需要传递backend="threading"

Parallel(n_jobs=3, backend="threading")(delayed(func)(i) for i in range(3))