From ad5c4350671c80596966456133cfaad1e96058c0 Mon Sep 17 00:00:00 2001
From: Erik Strand <erik.strand@cba.mit.edu>
Date: Wed, 8 May 2019 17:28:48 -0400
Subject: [PATCH] Generate different plots a different way

---
 compressed_sensing.py | 15 ++++++---------
 1 file changed, 6 insertions(+), 9 deletions(-)

diff --git a/compressed_sensing.py b/compressed_sensing.py
index 33e0e9d..360d74a 100644
--- a/compressed_sensing.py
+++ b/compressed_sensing.py
@@ -14,17 +14,14 @@ if __name__ == "__main__":
     n_samples = 1000
     sample_times = (sample_period / n_samples) * np.arange(n_samples)
     sample_values = sample_two_sins(f1, f2, sample_times)
-    fig, ax = plt.subplots()
-    ax.plot(sample_times, sample_values)
+    plt.plot(sample_times, sample_values)
+    plt.savefig("fig_a.png")
+    plt.close()
 
     sample_period = 0.02
     n_samples = 1000
     sample_times = (sample_period / n_samples) * np.arange(n_samples)
     sample_values = sample_two_sins(f1, f2, sample_times)
-    fig2, ax2 = plt.subplots()
-    ax2.plot(sample_times, sample_values)
-
-    plt.figure(fig.number)
-    plt.savefig("test.png")
-    plt.figure(fig2.number)
-    plt.savefig("test2.png")
+    plt.plot(sample_times, sample_values)
+    plt.savefig("fig_b.png")
+    plt.close()
-- 
GitLab