Comment créer automatiquement à l’aide de Python et de LaTeX un arbre de Bernoulli en probabilités ?
Je vais vous montrer l’utilisation d’une fonction Python que je viens d’écrire, et que les abonné·e·s trouveront en fin d’article.
LaTeX et Python: exemples d’arbres de Bernoulli
La fonction Python qui créée en LaTeX un arbre de Bernoulli
L’appel de la fonction nécessite obligatoirement deux paramètres: la hauteur de l’arbre et la probabilité du succès. Par exemple,
>>> bernoulli_tree(3,0.2)va automatiquement créer un fichier \(\LaTeX\) puis le compiler, pour ensuite afficher le PDF:
Comme on peut le remarquer, les points décimaux ont été remplacés par une virgule.
Une arbre de Bernoulli avec des probabilités fractionnaires: l’option “fraction”
Si l’on souhaite que les probabilités s’affichent sous forme fractionnaire, on utilisera l’option “fraction”:
>>> bernoulli_tree(3,0.2,fraction=True,espv=0.7)Remarquez ici la présence d’un autre paramètre : espv, qui permet d’espacer davantage les nœuds du dernier niveau afin que les fractions s’affichent correctement.
Si la probabilité du succès ne peut pas être écrite sous forme décimale, mais uniquement sous forme fractionnaire, on pourra écrire:
>>> bernoulli_tree(3,'1/3',fraction=True,espx=1,espv=0.8,espvinter=0.5)Un arbre moins large: l’option espx
>>> bernoulli_tree(3,0.2,espx=1,espv=0.7,espvinter=0.5)Remarquez ici la présence du paramètre espvinter qui désigne l’espace vertical entre les “groupes” de couples (succès,échec) au dernier niveau.
Par de nom à la racine
Si vous ne souhaitez pas que \(\Omega\) apparaisse à la racine, tapez:
>>> bernoulli_tree(3,'1/3',fraction=True,espx=1,espv=0.8,espvinter=0.5,root='')Changer les styles
On peut changer les styles TiKZ de tous les objets de l’arbre (branches, nœuds et probabilités):
>>> bernoulli_tree(3,0.2,fraction=True,espv=1.1,espvinter=0.5,stylebranche='line width=1pt,color=blue!50!black',stylenoeud='color=red',styleproba='midway,circle,scale=.5,text=red,draw=red,fill=red!20',filename='arbre-bernoulli-06')Ici, la seule limite est celle de votre imagination ! Vous pouvez utiliser tous les styles TiKZ que vous souhaitez pour personnaliser votre arbre. Alors, lâchez-vous !
Des fractions écrites “en ligne”
Jouer avec les espaces verticaux est quelques fois pénibles. Dans ce cas, on pourra utiliser la syntaxe suivante:
>>> bernoulli_tree(3,Fraction(3,7),filename='arbre-bernoulli-07')La fonction Python qui permet de construire en LaTeX un arbre de Bernoulli
Les paramètres utilisés pour la fonction Python permettant de construire en LaTeX un arbre de Bernoulli
- niveau : nombre entier désignant le niveau de l’arbre (nombre de répétitions de l’expérience de Bernoulli);
- proba: nombre (float) ou chaîne de caractères (de la forme ‘p/q’) ou objet type Fraction(p,q)
- succes = ‘S’: lettre désignant le succès. Par défaut, “S”
- espv = 0.5: espace vertical entre les nœuds “succès/échec”
- espvinter = 1: espace vertical (en cm) entre les blocs “succès/échec”
- espx = 2: espace horizontal (en cm) entre chaque niveau
- fraction = False: booléen pour signifier si les probabilités doivent être écrites sous forme fractionnaire ou non
- root = ‘$\Omega$’: nom de la racine (syntaxe \(\LaTeX\) mathématique ou pas)
- stylebranche = ”: style TiKZ des branches
- stylenoeud = ”: style TiKZ des nœuds (les événements “succès” et “échecs”)
- styleproba = ‘midway,fill=white,scale=.5’: style TiKZ des probabilités sur les branches
- filename = ‘arbre’: nom du fichier à sauvegarder et à compiler
Les modules
- ‘fraction’
- >>> pip install fractions ou conda install fractions
- ‘os’ et ‘os.path’ (pour manipulation système: sauvegarde, compilation)
Autres remarques
Il faut bien entendu au préalable avoir installé une distribution \(\LaTeX\).
La fonction Python
from fractions import Fraction
def bernoulli_tree( niveau,\
proba,\
succes = 'S',\
espv = 0.5,\
espvinter = 1,\
espx = 2,\
fraction = False,\
root = '$\\Omega$',\
stylebranche = '',\
stylenoeud = '',\
styleproba = 'midway,fill=white,scale=.5',
filename = 'arbre'):
# Stéphane Pasquet
# mathweb.fr
# 2022/07/02
# https://www.mathweb.fr/euclide/2022/07/02/latex-et-python-creer-un-arbre-de-bernoulli-automatiquement/
if fraction:
proba_num = Fraction(str(proba)).numerator
proba_denom = Fraction(str(proba)).denominator
if isinstance(proba,float):
proba_echec_num = Fraction(str(1-proba)).numerator
proba_echec_denom = Fraction(str(1-proba)).denominator
else:
proba_echec_num = Fraction(str(1-Fraction(proba))).numerator
proba_echec_denom = Fraction(str(1-Fraction(proba))).denominator
proba = '$\\frac{'+str(proba_num)+'}{'+str(proba_denom)+'}$'
probaechec = '$\\frac{'+str(proba_echec_num)+'}{'+str(proba_echec_denom)+'}$'
else:
probaechec = str(1-proba).replace(".",",")
proba = str(proba).replace(".",",")
tex = '\\documentclass{standalone}\n'
tex += '\\usepackage{tikz}\n'
tex += '\\usetikzlibrary{calc}\n'
tex += '\\setlength{\\parindent}{0pt}\n'
tex += '\\begin{document}\n'
tex += '\\begin{tikzpicture}\n'
tex += '\\tikzstyle{branche} = [ '+stylebranche+' ]\n'
tex += '\\tikzstyle{noeud} = [ '+stylenoeud+' ]\n'
tex += '\\tikzstyle{proba} = [ '+styleproba+' ]\n'
level = niveau
echec = '\\overline{' + succes + '}'
# placement des noeuds
while level != 0:
if level == niveau:
y = 0
for i in range(2**level):
if i%2 == 0:
tex += '\\node[noeud] (N' + str(level) + '-' + str(i) + ') at ({'+str(espx)+'*'+str(level)+'},'+str(y)+') {$'+succes+'$};\n'
y -= espv
else:
tex += '\\node[noeud] (N' + str(level) + '-' + str(i) + ') at ({'+str(espx)+'*'+str(level)+'},'+str(y)+') {$'+echec+'$};\n'
y -= espvinter
else:
for i in range(2**level):
if i%2 == 0:
tex += '\\node[noeud] (N' + str(level) + '-' + str(i) + ') at ($0.5*(N'+str(level+1)+'-'+str(2*i)+')+0.5*(N'+str(level+1)+'-'+str(2*i+1)+')+(-'+str(espx)+',0)$) \
{$'+succes+'$};\n'
else:
tex += '\\node[noeud] (N' + str(level) + '-' + str(i) + ') at ($0.5*(N'+str(level+1)+'-'+str(2*i)+')+0.5*(N'+str(level+1)+'-'+str(2*i+1)+')+(-'+str(espx)+',0)$) \
{$'+echec+'$};\n'
level -= 1
# tracé des branches
tex += '\\node[left,noeud] at ($0.5*(N1-0)+0.5*(N1-1)+(-'+str(espx)+',0)$) {'+root+'};\n'
tex += '\\draw[branche] ($0.5*(N1-0)+0.5*(N1-1)+(-'+str(espx)+',0)$) -- (N1-0) node[proba] {'+proba+'};\n'
tex += '\\draw[branche] ($0.5*(N1-0)+0.5*(N1-1)+(-'+str(espx)+',0)$) -- (N1-1) node[proba] {'+probaechec+'};\n'
for lev in range(1,niveau):
for i in range(2**lev):
tex += '\\draw[branche] (N'+str(lev)+'-'+str(i)+'.east) -- (N'+str(lev+1)+'-'+str(2*i)+') node[proba] {'+proba+'};\n'
tex += '\\draw[branche] (N'+str(lev)+'-'+str(i)+'.east) -- (N'+str(lev+1)+'-'+str(2*i+1)+') node[proba] {'+probaechec+'};\n'
tex += '\\end{tikzpicture}\n'
tex += '\\end{document}'
# sauvegarde du fichier LaTeX
from os.path import isfile
from os import remove, system
if isfile(filename+".tex"):
remove(filename+".tex")
fichier = open(filename+".tex","x")
fichier.write(tex)
fichier.close()
# compilation PdfLaTeX dans le répertoire courant
cmd = "pdflatex --shell-escape -synctex=1 -interaction=nonstopmode "+filename+".tex"
system(cmd)
cmd = "START "+filename+".pdf"
system(cmd)
