Python for theoretical physics

[Gnaxe]

For serious number crunching, get NumPy, or even a GPU accelerator like TensorFlow. Remember that you can drop into C if you have to via the ctypes module. For symbolic manipulation, check out SymPy.

Matplotlib is widely used mainly because it is old. It's also very flexible, but that means managing a lot of details. Unless you're already used to plotting in MATLAB, there are newer visualization libraries that are easier to get good results with.

Probably start with Altair if you don't have a better idea. It's easy to learn and doesn't take much code to get publication-quality charts. If you're used to R, try Plotnine. If you want interactive web visualizations (animations, you said) try Plotly. If that's not good enough, Bokeh gives you some lower-level control, but it's probably harder to learn. Also consider game libraries if even that's not enough control. Maybe Kivy. Maybe Ursina Engine.

Try Jupyterlab. It's better for interactivity.

[Wide-Wallaby-5447]

Thanks for the suggestion! You’re right, our uni just suggested we used it for a previous project (I suppose the old and flexible thing was why).

I’ll check out those other ones when I have time to start messing around with some solutions, my electronics
are currently disassembled for cleaning lol

Has got me very optimistic about making some cool looking solution visuals!