pythonマルチプロセッシングstarmapプログレスバー

Question

私はワーカー関数でいくつかのargsが必要なので、私はstarmapを使いますが、どのように進捗状況をtqdmで表示できますか？

from itertools import repeat 
from multiprocessing import Pool 

def func(i, a, b, c): 
    print(i, a, b, c) 

if __name__ == '__main__': 
    pool = Pool(thread_num) 
    a, b, c = 'a', 'b', 'c' 
    pool.starmap(func, zip(range(100), repeat(a), repeat(b), repeat(c))) 
    pool.close() 
    pool.join() 

私はどのようにしてtqdmにプレビューを表示できますか？

Answer 1

他のプロセスから渡されたシグナルを監視し、tqdmを更新するプロセスを作成する必要があります。あなたのための最小の例：

from multiprocessing import Queue, Pool, Process 

def listener(q, num): 
    tbar = tdqm(total = num) 
    for i in iter(q.get, None): 
     tbar.update() 
    tbar.close() 

def worker(q): 
    do something. 
    queue.put(1) 

if __name__ == "__main__": 
    .... 
    q.put(None) #when you finish your work, put a None signal to finish the listener.