update plot

amanmdesai · Sep 27, 2023 · 03b3709 · 03b3709
1 parent 5d742bf
commit 03b3709
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/pymcabc/plotting.py b/pymcabc/plotting.py
@@ -13,10 +13,12 @@ class for plotting data
     """
 
     def plot(data, key):
-        plt.hist(data, bins=40, color=None)
+        fig, ax = plt.subplots()
+        ax.cla()
+        ax.hist(data, bins=40, color=None)
         label = key.replace("_", " ")
-        plt.xlabel(label)
-        plt.ylabel("Events")
+        ax.set_xlabel(label)
+        ax.set_ylabel("Events")
         plt.savefig(key + ".png")
 
     def file(filename="ABC_events.root"):