Skip to content

반복문 진행상황 시각화

tqdm 라이브러리로 반복문의 진행상황 시각화하기


tqdm 라이브러리 활용하기

Python의 반복문이 얼마나 진행이 되었는지 궁금할 때가 있다. 이 때는 아래와 같이 tqdm 라이브러리를 활용하면 손쉽게 파악할 수 있다.

from tqdm import tqdm

tmp = [i for i in tqdm(range(10 ** 7))]

중첩 반복문에 tdqm 사용하기

중첩 반복문에서 tqdm을 제대로 사용하려면 아래와 같이 leave 파라미터 설정이 필요하다.

import time

from tqdm import tqdm

for i1 in tqdm(range(5)):
    for i2 in tqdm(range(5), leave=False):
        for i3 in tqdm(range(100), leave=False):
            time.sleep(0.00001)

pandas에 tqdm 사용하기

pandas DataFrame의 iterrows나 itertuples 메서드를 활용할 때는 total 파라미터 전달이 필요하다.

import time

import pydataset as pds
from tqdm import tqdm

data = pds.data('iris')

for idx, row in tqdm(data.iterrows(), total=data.shape[0]):
    time.sleep(0.0001)

Reference