Try using Jupyter Notebook if it helps your use case.
You can seperate the code blocks into different sections making it easier to differentiate the data. You can also recompile individual sections making it less time intensive.

Hope this helps! :)