% Dans le préambule
% -----------------
\usepackage{pythontex}
% -----------------

On définit une fonction

\begin{pyconsole}
import math
def aire(r):
  return math.pi*r**2

aire(4)
\end{pyconsole}