Get (a posteriori) where the RunTime warnings happened in a program Python

Question

I have a big dataset of 4D brain image data. I\'m iterating in a 3d for-loop within images to run some processes in each voxel (e.g., non-linear fitting + MCMC). The for-loop is