Vscode has this built in: https://code.visualstudio.com/docs/python/jupyter-support-py