I also recommend jupyterlab for a similar experience. I was a hardcore vim terminal guy for 17 years, but now I won’t be going back.
The ability to interactively develop functions in almost any language inside the same environment had been revolutionary for my productivity given my short attention span.