My team writes a lot of Python code in PyCharm to do quick simulations and calculations for our mechanical engineering work.
Since it looks like the kind of code that you write (quick simulations and calculations) isn't full blown application code that these sorts of IDEs are designed for (as in, code for some application software that needs to be distributed and maintained), you might find tools used by data scientists / analysts better suited for the kinds of work that you do.
Look into Jupyter Notebook (or the newer JupyterLab) - it allows you to intersperse text, images and runnable Python code (code that can be executed right there in the notebook itself, it can generate outputs, diagrams, etc.) It's not as powerful as an IDE in terms of development features, but since you are primary dealing in mechanical engineering work, you might not be using many of those features anyway. Can't say for sure if this is what solves your needs, but it's worth checking out.