tqdm 이란 - 파이썬 상태 바 라이브러리

tqdm 이란

dataloader을 통해 data를 download할 때, 상태바를 넣어두는 경우가 있다.

from tqdm import tqdm

tqdm이란, 상태 진행(progress) 을 보여주는 파이썬 라이브러리 이다

 

코드 예시

1. tqdm(iterable)

tqdm(iterable) 을 해주면, iterable 내에서의 상태 진행률을 알려준다.

from tqdm.notebook import tqdm

for i in tqdm(range(100)):
     i+=i

 

2. total = tqdm(iterable)

from tqdm.notebook import tqdm
import time

total = tqdm(range(100))
for i in total:
    time.sleep(0.1)

3. trange(i)

trange(i) 는 tqdm(range(i)) 와 같은 기능을 가지고 있다.

from tqdm.notebook import tnrange
import time

#trange(i) is a special optimised instance of tqdm(range(i)):
for i in tnrange(5, desc = "first"):
    for j in tnrange(100, desc="second"):
        time.sleep(0.1)

4. set_description

set_description 을 통해서 iteration의 어느 부분이 진행 중인지 확인할 수 있다.

from tqdm.notebook import tqdm
import time

total = tqdm(["a", "b", "c", "d"], desc = "abcd")
for char in total:
    time.sleep(10)
    total.set_description("Processing %s" % char)

맨 처음, desc 부분의 내용과 함께 0%의 진행률을 보인다.

그러다 10초 후, (time.sleep(10)) Processing a로 바뀌며 a 부분이 진행됨을 알려줌과 동시에 25%의 진행률을 보여준다.

모든 진행이 끝나면 다음과 같이 표시 된다.

tqdm 깃허브(github)

다음의 링크를 참고하여 글을 작성했으며, 더 자세한 내용은 아래의 링크를 통해 확인할 수 있다.

github.com/tqdm/tqdm

tqdm/tqdm