파이썬 다중 처리 starmap progressbar

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.