我在这个带有 pool.map 的文件中做错了什么导致什么都没有出现，我必须重新启动 shell？

Question

在以下使用多处理的 python 文件中我遗漏了什么或做错了什么？当我运行它时，没有任何反应，我必须重新启动 shell!

def f(x):
    lo=0
    for i in range(x):
            lo+=i
    return(lo)
from multiprocessing import Pool
def f_parallel(x1,x2,x3,x4):
    with Pool(processes=4) as pool:
        resulto_parallel=pool.map(f,[x1,x2,x3,x4])
    return(resulto_parallel)
f_parallel(1,2,3,4)

这是我运行时发生的情况的屏幕截图。

等待一段时间后，我重新启动 shell。

Answer 1

您需要使用 'module name' 成语：

def f(x):
    lo = 0
    for i in range(x):
            lo += i
    return lo

from multiprocessing import Pool

def f_parallel(x1, x2, x3, x4):
    with Pool(processes=4) as pool:
        return pool.map(f,[x1, x2, x3, x4])

if __name__ == '__main__':
    print(f_parallel(1, 2, 3, 4))

输出：

[0, 1, 3, 6]

没有上面的 if __name__ == '__main__': 部分，multiprocessing 重新读取模块（执行它）并进入无限循环。

Answer 2

引用自the multiprocessing docs：

one should protect the “entry point” of the program by using if __name__ == '__main__'

在你的情况下，这看起来像：

from multiprocessing import Pool

def f(x):
    lo = 0
    for i in range(x):
            lo += i
    return lo

def f_parallel(x1, x2, x3, x4):
    with Pool(processes=4) as pool:
        resulto_parallel = pool.map(f, [x1, x2, x3, x4])
    return resulto_parallel 

if __name__ == '__main__':    
    print(f_parallel(1, 2, 3, 4))

关键是生成的进程将导入您的 python 脚本，如果您不保护对 f_parallel(1,2,3,4) 的调用，将导致子进程的无限生成。

注意：还加了一个print这样函数调用的结果实际上会放到stdout

我在这个带有 pool.map 的文件中做错了什么导致什么都没有出现，我必须重新启动 shell？

What am I doing wrong in this file with pool.map which causes nothing appearing and I have to restart the shell?

python

python-multiprocessing

python-3.7