If you want to display python code and its output as LaTeX, try latexify_py. With this, you can print python code as a LaTeX expression and make your code more interpretable. What’s more, it can also generate LaTeX code for python code. This saves plenty of time and effort of manually writing the expressions in LaTeX.