From 3548aa15dff8390d9b3f44a571198f765abf590a Mon Sep 17 00:00:00 2001
From: Erik Strand <erik.strand@cba.mit.edu>
Date: Wed, 8 May 2019 17:26:51 -0400
Subject: [PATCH] Generate two plots
---
compressed_sensing.py | 30 ++++++++++++++++++++++++++++++
1 file changed, 30 insertions(+)
create mode 100644 compressed_sensing.py
diff --git a/compressed_sensing.py b/compressed_sensing.py
new file mode 100644
index 0000000..33e0e9d
--- /dev/null
+++ b/compressed_sensing.py
@@ -0,0 +1,30 @@
+import numpy as np
+import matplotlib.pyplot as plt
+
+
+def sample_two_sins(f1, f2, sample_times):
+ sample_rads = 2 * np.pi * sample_times
+ return np.sin(f1 * sample_rads) + np.sin(f2 * sample_rads)
+
+if __name__ == "__main__":
+ f1 = 697 # Hz
+ f2 = 1209 # Hz
+
+ sample_period = 0.01
+ 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)
+
+ 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")
--
GitLab