diff --git a/compressed_sensing.py b/compressed_sensing.py index b9620292de4c9500feff43f024e128994e0bcb9f..39edcd5c9315c806ca53e6fd02a3b27ce751e420 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()