Add "Print_mat"

2020-04-29 09:28:12 +02:00 · 2020-04-29 09:28:12 +02:00 · f2730b39c6
commit f2730b39c6
parent 50cbfce858
1 changed files with 10 additions and 0 deletions
--- 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()