book_generator/gantt_parser.py

33 lines
992 B
Python

import pandas as pd
import matplotlib.pyplot as plt
# Lire le fichier CSV
df = pd.read_csv('intrigues.csv')
# Créer un diagramme de Gantt
fig, ax = plt.subplots(figsize=(10, 6))
# DéFinir les valeurs de Début et de Fin pour chaque tâche
for i, row in df.iterrows():
ax.plot([row['Début'], row['Fin']], [i, i], 'b-')
ax.plot([row['Début'], row['Début']], [i-0.1, i+0.1], 'bo')
ax.plot([row['Fin'], row['Fin']], [i-0.1, i+0.1], 'ro')
# DéFinir les étiquettes pour les tâches
ax.set_yticks(range(len(df)))
ax.set_yticklabels(df['Intrigue'])
# DéFinir les étiquettes pour les valeurs
ax.set_xticks(range(int(df['Début'].min()), int(df['Fin'].max())+1))
ax.set_xticklabels(ax.get_xticks())
ax.plot([row['Début'], row['Fin']], [i, i], 'b-', linewidth=2)
# Ajouter un titre au diagramme
ax.set_title('Intrigues')
# Afficher le diagramme
#plt.show()
#plt.figure(figsize=(16, 9))
# Code pour tracer votre graphique
plt.savefig("graphique_gantt_intrigues.png", dpi=72)