[Pandas] Pandas에서 tqdm 사용하기

tqdm을 사용할 때는 tqdm.tqdm이나 tqdm.trange를 사용하여 for 문에 적용한다. 그러나 pandas apply를 사용할 때 시간이 오래 걸리지만 tqdm을 사용하기 어렵다. 이 경우 어떻게 할 수 있을까?

 

이를 지원해주는 것이 바로 tqdm.pandas()이다. 

import pandas as pd 
import tqdm.tqdm as tqdm 
from tqdm import trange 

tqdm.pandas()


with trange(len(x)) as tr:
	for i in tr:
    	print(x[i])
        

for i in tqdm(range(10), desc = 'training...'):
	print(x[i])


dataframe.progress_apply(lambda x: x.split())

위 코드는 유사 코드로 작성했다. 사용하고 싶은 방법에 따라 다양한 방법으로 사용하면 된다.

 

from tqdm import tqdm, tqdm_pandas 
tqdm_pandas(tqdm())

tqdm이 예전 버전인 경우 즉, 4.8 버전 이하인 경우에는 위 코드를 작성해야만 사용이 가능하다.