tqdm and numpy vectorize

Question

I am using a np.vectorize-ed function and would like to see the progress of the function with tqdm. However, I have not been able to figure out how to do this.

All the suggestions I have found relate to converting the calculation into a for-loop, or into a pd.DataFrame.

Answer 1

I finally found a method that works to get the tqdm progress bar to update with a np.vectorize function. I wrap the vectorize function using a with

with tqdm(total=len(my_inputs)) as pbar:
    my_output = np.vectorize(my_function)(my_inputs)

in my_function() I then add the following lines

global pbar
pbar.update(1)

and voila! I now have a progress bar that updates with each iteration. Only slight performance dip on my code.

Note: when you instantiate the function it might complain that pbar is not yet defined. Simply put a pbar = 0 before you instantiate, and then the function will call the pbar defined by the with

Hope it helps everyone reading here.