python - 如何将函数的返回值附加到池中的列表中？

Question

我很好奇是否可以将函数 f(x,y) 的返回值附加到例如列表中。

所以我有这个:

平行:

 def download_unit_LVs(self, cislo_uzemia):
     pool = Pool(number_of_workers)
     for cislo in cisla:
         pool.apply_async(self.download_cislo, args=(cislo_uzemia,cislo))
     pool.close()
     pool.join()
     self.manager.commit()

这是我并行运行 self.download_cislo 方法的方式，但问题是，它返回一个值，我必须将其附加到结果列表中。

怎么做？

顺序:

 def download_unit_LVs(self, cislo_uzemia):
        results = []
        for cislo in cisla:
            results.append(self.download_cislo(cislo_uzemia,cislo))
        self.manager.commit()

Answer 1

pool.apply_async 调用可以通过 callbackfunction .这foo 函数完成时将调用回调函数，并将传递了 foo 返回的值。注意返回值must be picklable ，因为进程之间的通信是通过队列完成的。

import multiprocessing as mp

def foo(x):
    return x * x

if __name__ == '__main__':
    result = []
    pool = mp.Pool()
    for i in range(100):
        pool.apply_async(foo, args=(i, ), callback=result.append)
    pool.close()
    pool.join()
    print(result)
    # [0, 1, 4, 9, 16, ... 9409, 9604, 9801]