머신러닝&딥러닝/Python

tqdm을 사용하여 처리 진행상황 확인하는 방법(for문, apply문)

e냥냥 2022. 9. 14. 14:23
728x90

데이터 전처리를 하다보면 반복작업을 하는 경우가 생기고 데이터 양이 많으면 어느 정도 진행되었는지,

언제 끝나는지 모르는 경우도 생깁니다. 그럴때 tqdm를 사용하여 현재 진행상황을 확인할 수 있습니다.

 

먼저 tqdm 패키지를 conda install tqdm으로 설치하고, 불러옵니다. 

from tqdm import tqdm

 

1. for문 적용

for문에 적용하는 경우 in 우측 반복할 변수에 씌워주면 됩니다.

for x in tqdm(range(10000000)):
    x2 = x^2

 

2. apply문 적용

apply는 주로 pandas row를 처리하는 경우 사용하는데 이때는 tqdm.pandas()를 선언해준 후

progress_apply 함수를 이용해야합니다. 

tqdm.pandas()
x['columns2'] = x['columns1'].progress_apply(lambda x : x^2)

 

 

 

 

감사합니다 :)

728x90
loading