Just curious: That code could be run by just plain `#+begin_src python`. What does using Jupyter gain you?
Nothing inherent to Jupyter (it is not like Jupyter notebooks vs. Python REPL), just some niceties:
- low latency (it feels more responsive)
- interactive output (tqdm works)
- how Python objects are displayed by default
Essentially a good implementation of Python code blocks would reimplement half of Jupyter anyway.
Here's an example emacs configuration:
After that, you could execute jupyter code block (Org Babel): You can use latex in org-mode, generate pictures, get nice git diff, export it to pdf, etc.