Comment by pizza

You could use a jupyter notebook for this

Side-note: is the inverse of this available? E.g. open a jupyter notebook from in the middle of a python script