如何从multiprocessing.Process获取返回值?

时间:2019-08-13 17:30:29

标签: python multiprocessing

我正在尝试在代码中实现并行化,但是我无法从multiprocessing.Process获得结果。我有一个看起来像这样的代码:

if __name__ == '__main__':
    labels = ['neutral', 'bad', 'good']
    procs = []
    for i in labels:
        p = Process(target = process, args = (i, ))
        procs.append(p)
        p.start()
    for proc in procs:
        proc.join()

函数process加载NLTK语料库,对文档进行一些转换并返回一个列表。我需要使用不同的参数调用此函数三次,以处理三个文件夹。所以我正在考虑并行运行它。我想使用labels中的键将返回列表添加到字典中。结果应如下所示:

result = {'neutral': [returned_list], 'bad': ....

我已经读过QueuePipe的书,但是我不知道该如何使用它们,因此我很乐意提供帮助。

1 个答案:

答案 0 :(得分:1)

您可以这样尝试

import multiprocessing

def action(data):
    return {data:["computed_action"]}

labels = ['neutral', 'bad', 'good']
proc_count = 5
result = multiprocessing.Pool(proc_count).map(action,labels)
print(result)

输出

[{'neutral': ['computed_action']}, {'bad': ['computed_action']}, {'good': ['computed_action']}]