diff --git a/Divers/Affichage.py b/Divers/Affichage.py index 6896a10..64f0086 100644 --- a/Divers/Affichage.py +++ b/Divers/Affichage.py @@ -61,3 +61,13 @@ def Print_plt(Mat, fichier, rep=".\\", afficher=False, save=True): if afficher: plt.show() plt.close() + + +def Print_mat(y, fichier, rep=".\\", afficher=False, save=True): + heatmap = plt.pcolor(y) + plt.colorbar(heatmap) + if save: + plt.savefig(rep+fichier+".png") + if afficher: + plt.show() + plt.close()