It's not a GUI library, but Jupyter was pretty much made for the kind of mathematical/scientific exploratory programming you're interested in doing. It's not the right tool for making finished products, but is intended for creating lab notebooks that contain executable code snippets, formatted text, and visual output together. Given your background experience and the libraries you like, it seems like it'd be right up your alley.