EDIT: Googled and answered my own question. Here are docs describing the feature: https://code.visualstudio.com/docs/python/jupyter-support-py
And a video demoing what you are describing: https://www.youtube.com/watch?v=lwN4-W1WR84
EDIT: Googled and answered my own question. Here are docs describing the feature: https://code.visualstudio.com/docs/python/jupyter-support-py
And a video demoing what you are describing: https://www.youtube.com/watch?v=lwN4-W1WR84