Using Wolfram Mathematica for this is more satisfying than it should be
https://www.wolframcloud.com/obj/agam.brahmawolfram/Published/AOC%202021.nb
It was shockingly easy to do all this.
- REPL/Notebook glory with fast-n-quick editing and modification
- easy switching between text and code cells
- easy prototyping
- easy plotting
I’m sure anything that requires “running systems” would be a bad fit, but most cases where you’d think “I need a real programming language here”, Wolfram Language is more than adequate.