如何将函数的返回值附加到Pool中的列表中?

时间:2015-12-30 11:54:15

标签: python multithreading python-multithreading pool

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

所以我有这个:

并行

 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()

2 个答案:

答案 0 :(得分:5)

pool.apply_async来电可以传递callback function。该 当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]

答案 1 :(得分:1)

如果结果应该以列表结尾,请使用map()的{​​{1}}方法:

Pool

def download_unit_LVs(self, cislo_uzemia): pool = Pool(number_of_workers) results = pool.map(partial(self.download_cislo, cislo_uzemia), self.cisla) self.manager.commit() 来自partial()模块。