[데이터 사이언스/문법] Python 라이브러리 - tqdm 이란?

2024. 8. 14. 19:16데이터 사이언스

tqdm은 Python에서 진행률 표시줄을 간편하게 추가할 수 있는 라이브러리입니다.
루프의 진행 상황을 시각적으로 보여주어 작업이 얼마나 진행되었는지,
얼마나 남았는지를 쉽게 파악할 수 있습니다.

 

기본 예제

tqdm을 사용하여 간단한 루프의 진행 상황을 출력하는 예제

from tqdm import tqdm
import time

# 예제 작업: 100번의 반복 수행, 각 반복마다 0.1초 대기
for i in tqdm(range(100), desc="Processing"):
    time.sleep(0.1)


출력 결과

이 코드를 실행하면 터미널이나 콘솔에 다음과 같은 진행률 표시줄이 나타납니다.

Processing: 100%|████████████████████████████████| 100/100 [00:10<00:00,  9.99it/s]

  • Processing: desc 파라미터에 의해 설정된 진행률 표시줄의 설명입니다.
  • 100%: 현재 작업이 완료된 백분율입니다.
  • ████████████████████████████████: 시각적인 진행률 표시줄입니다.
  • 100/100: 현재 반복 횟수와 총 반복 횟수입니다.
  • [00:10<00:00, 9.99it/s]: 진행된 시간, 남은 시간, 초당 반복 횟수를 나타냅니다.

다른 예제: 데이터프레임 처리

tqdm은 다양한 곳에 적용할 수 있습니다. 예를 들어, 데이터프레임의 행을 처리할 때도 사용할 수 있습니다

import pandas as pd
from tqdm import tqdm

# 샘플 데이터프레임 생성
data = {'col1': range(1000), 'col2': range(1000)}
df = pd.DataFrame(data)

# tqdm 적용
for index, row in tqdm(df.iterrows(), total=df.shape[0], desc="Processing rows"):
    # 각 행(row)에 대해 작업 수행
    pass


출력 결과

이 코드를 실행하면 터미널이나 콘솔에 다음과 같은 진행률 표시줄이 나타납니다

Processing rows: 100%|████████████████████████████████████████| 1000/1000 [00:01<00:00, 800.00it/s]