Marimo notebook would fit nicely here for interactive examples where you could play with values and see computations in real time, though it probably lacks in formatting capabilities and quite heavy.