如何在多处理中使用此功能？

Question

我想在多处理中应用这个函数：

import multiprocessing
import numpy as np

a = np.random.randint(1,4, size=(10,10,10))

def function(input_field,[num1,num2,num3]):
    pass

if __name__ == '__main__':
    pool = multiprocessing.Pool(processes=4)
    test_list = [(a,[27.5, 27.5, 25]), (b,[27.5, 27.5, 25]), (c,[27.5, 27.5, 25]), (d,[27.5, 27.5, 25])]
    results = pool.map(function, test_list)
    pool.close()
    pool.join()    

如何将function(input_field,[num1,num2,num3])应用于多处理？

Answer 1

根据test_list的结构，您有 2 个选项：

1. 修改您的函数以支持单个参数并将其解压缩到函数中并继续使用map ：

    def function(arg): input_field, l = arg

2. 使用starmap并修复您的函数定义，因为test_list的结构是starmap期望获得的：

    def function(input_field, l): ... results = pool.starmap(function, test_list)