Have you seen Python Interactive? There are no separate editors for each cell, but you edit a single file and use #%% to define the boundaries between cells. https://code.visualstudio.com/docs/python/jupyter-support-py