From f2730b39c67da4f378d48ddd11d5b4a8252d2057 Mon Sep 17 00:00:00 2001
From: Paul-Corbalan <paulcorbalan@yahoo.fr>
Date: Wed, 29 Apr 2020 09:28:12 +0200
Subject: [PATCH] Add "Print_mat"

---
 Divers/Affichage.py | 10 ++++++++++
 1 file changed, 10 insertions(+)

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()