Comment by timfi

This looks cool. If you want to delve deeper intob symbolic computation you can have a look at sympy [0]. It seems to be the "defacto standard" when it comes to Python libraries for this.

0: https://www.sympy.org/en/index.html