[python] tqdm 사용법

1. 설명

tqdm은 파이썬에서 진행상황을 알려주는 프로그레스바 (progress bar)를 생성할 때 쓰이는 라이브러리이다.

 

2. 사용법

tqdm은 리스트와 유사한 인스턴스를 감싼 후 for문과 함께 쓰인다.

from tqdm import tqdm
for i in tqdm(range(10)):
	i
    
for i in [1,2,3]:
	i
    
for i in set(['a','b','c']):
	i

 

3. 커스터마이징

tqdm의 장점으로는 각 과정 (tick) 때마다 프로그레스바에 사용자가 원하는 값을 출력할 수 있다는 것이다.

이것은 딥러닝을 할 때 loss값을 표기할 때 유용하다.

from tqdm import tqdm
import time

progress_bar=tqdm(range(100))

for i in progress_bar:
    time.sleep(0.1)
    progress_bar.set_description(f'Loss :{i}')
    
Loss :49:  50%|████████████████████████████████████████████████████        | 50/100 [00:08<00:05,  9.92it/s