\documentclass[10pt, b5paper]{article} \usepackage{fancyhdr,amssymb,amsmath} \usepackage{pyluatex} \begin{python} import math import random import numpy as np np.random.seed(0) \end{python} \begin{document} \begin{python} a = 7 \end{python} \pyif{a == 7}{\begin{center} a \end{center}}{a \par b} \pyif{a == 8}{\begin{center} a \end{center}}{a \par b} \end{document}