For quick reference here's the tutorial: https://orgmode.org/worg/org-contrib/babel/intro.html

Once you have it set up you can have Python code in the file like this:

#+begin_src python
# python code here
#+end_src