From 5665812e05bb3df420c279eced9e5dd73a365978 Mon Sep 17 00:00:00 2001
From: Erik Strand <erik.strand@cba.mit.edu>
Date: Wed, 8 May 2019 17:48:09 -0400
Subject: [PATCH] Plot recovered data

---
 compressed_sensing.py | 7 +++++++
 1 file changed, 7 insertions(+)

diff --git a/compressed_sensing.py b/compressed_sensing.py
index b962029..39edcd5 100644
--- a/compressed_sensing.py
+++ b/compressed_sensing.py
@@ -39,3 +39,10 @@ if __name__ == "__main__":
     plt.plot(np.arange(n_samples), dct)
     plt.savefig("fig_b.png")
     plt.close()
+
+    # Part (c)
+    inverse_dct_matrix = np.transpose(dct_matrix)
+    recovered_sample_values = np.matmul(inverse_dct_matrix, dct)
+    plt.plot(sample_times, recovered_sample_values)
+    plt.savefig("fig_c.png")
+    plt.close()
-- 
GitLab