Python에서 Progress bar로 진행상황 표현하기

코드를 돌리는데 먼가 진행되는 것 같은데 확인을 할 수 없으니까 저는 많이 답답하더라고요.

 

그래서 오늘은 진행상황을 알 수 있는 progress bar를 표현하는 방법에 대해 알아보도록 하겠습니다.

 

tqdm 이라는 모듈을 사용하면 쉽게 구현이 가능합니다. 

 

먼저 tqdm을 설치합니다.

 

pip install tqdm 

 

그 다음에 코드문에서 tqdm을 import 합니다. 

from tqdm import tqdm

 

그리고 반복문이 들어가는 곳에 tqdm을 사용해주면 됩니다. 

 

간단하게 range를 사용한다면 이렇게 사용해주면 되고, 

 

from tqdm import tqdm

for i in tqdm(range(10)) : 
	print(i)

 

만약 dataframe 에서 iterrows를 사용한다면 다음과 같이 활용하시면 됩니다. 

 

from tqdm import tqdm

sample_df = pd.DataFrame(
    [
    {'meta1' : 'A', 'meta2': 'B', 'para' : '1', 'val1' : 'AA', 'val2' : 'AB' } ,
    {'meta1':'A', 'meta2':'B',  'para': '2', 'val1':'BA', 'val2':'BB'},
    {'meta1':'A', 'meta2':'B',  'para': '3', 'val1':'AA', 'val2':'BB'},
    {'meta1': 'C', 'meta2': 'D', 'para': '1', 'val1': 'CC', 'val2': 'CD'},
    {'meta1': 'C', 'meta2': 'D', 'para': '2', 'val1': 'DC', 'val2': 'DD'},
    {'meta1': 'C', 'meta2': 'D', 'para': '3', 'val1': 'CC', 'val2': 'DD'},
    {'meta1': 'C', 'meta2': 'D', 'para': '3', 'val1': 'CC', 'val2': 'DD'}
    ])


for i, row in tqdm(sample_df, total=sample_df.shape[0]):
	print(i)
    print(row)