From fdd324dc2461c160e76f5f084507d1802615cce5 Mon Sep 17 00:00:00 2001
From: Erik Strand <erik.strand@cba.mit.edu>
Date: Wed, 8 May 2019 18:03:15 -0400
Subject: [PATCH] Generate a subset of samples
---
compressed_sensing.py | 13 +++++++++++++
1 file changed, 13 insertions(+)
diff --git a/compressed_sensing.py b/compressed_sensing.py
index 39edcd5..775c6bb 100644
--- a/compressed_sensing.py
+++ b/compressed_sensing.py
@@ -46,3 +46,16 @@ if __name__ == "__main__":
plt.plot(sample_times, recovered_sample_values)
plt.savefig("fig_c.png")
plt.close()
+
+ # Part (d)
+ n_subsamples = 100
+ np.random.seed(7250147)
+ subset_indices = np.arange(n_samples)
+ np.random.shuffle(subset_indices)
+ subset_indices = subset_indices[:n_subsamples]
+ subset_indices = np.sort(subset_indices)
+ subset_sample_times = sample_times[subset_indices]
+ subset_sample_values = sample_values[subset_indices]
+ plt.plot(subset_sample_times, subset_sample_values)
+ plt.savefig("fig_d.png")
+ plt.close()
--
GitLab