Python3 多線程函數(shù)返回值

第一種寫法,多進(jìn)程的函數(shù)是沒有返回值的,需要放在Queue中來(lái)取值

import multiprocessing as mp

def job(char, num, q):
    q.put([char] * num)

if __name__ == "__main__":
    q = mp.Queue()
    p1 = mp.Process(target=job, args=("a", 3, q))
    p2 = mp.Process(target=job, args=("b", 4, q))
    p1.start()
    p2.start()
    p1.join()
    p2.join()
    print(q.get())
    print(q.get())

join()用于等待子進(jìn)程退出,輸出為

['b', 'b', 'b', 'b']
['a', 'a', 'a']

第二種寫法,線程池

import multiprocessing as mp

def job(char, num):
    return [char] * num

if __name__ == "__main__":
    with mp.Pool(processes=mp.cpu_count() - 1) as pool:
        p1 = pool.apply_async(job, ('a', 3))
        p2 = pool.apply_async(job, ('b', 4))
        print(p1.get(timeout=1))
        print(p2.get(timeout=1))

輸出類似

['a', 'a', 'a']
['b', 'b', 'b', 'b']

其中這個(gè)pool.join()要不要加好像不要緊,參考這個(gè)回答when-should-we-call-multiprocessing-pool-join
第三種寫法:

import multiprocessing as mp

def job(char, num):
    return [char] * num

if __name__ == "__main__":
    args = [("a", 4), ("b", 3)]
    with mp.Pool(processes=mp.cpu_count() - 1) as pool:
        results = pool.starmap(job, args)
    print(results)

返回值

[['a', 'a', 'a', 'a'], ['b', 'b', 'b']]

非常類似的寫法:

import multiprocessing as mp

def job(char, num):
    return [char] * num

if __name__ == "__main__":
    args = [("a", 4), ("b", 3)]
    with mp.Pool(processes=mp.cpu_count() - 1) as pool:
        results = pool.starmap_async(job, args)
        print(results.get(timeout=1))

參考文獻(xiàn)

stackoverflow: python multiprocessing pool map for multiple arguments
https://docs.python.org/3/library/multiprocessing.html

?著作權(quán)歸作者所有,轉(zhuǎn)載或內(nèi)容合作請(qǐng)聯(lián)系作者
【社區(qū)內(nèi)容提示】社區(qū)部分內(nèi)容疑似由AI輔助生成,瀏覽時(shí)請(qǐng)結(jié)合常識(shí)與多方信息審慎甄別。
平臺(tái)聲明:文章內(nèi)容(如有圖片或視頻亦包括在內(nèi))由作者上傳并發(fā)布,文章內(nèi)容僅代表作者本人觀點(diǎn),簡(jiǎn)書系信息發(fā)布平臺(tái),僅提供信息存儲(chǔ)服務(wù)。

友情鏈接更多精彩內(nèi)容