I use jupytext for this. For the cost of some header boilerplate I have python code that is seamlessly mirrored as a notebook.