exercises/exerciseA/plot.py

@@ -2,13 +2,15 @@ import os
 import numpy as np
 import matplotlib.pyplot as plt
 
+
 # IO: This loads the timings for you 
 threads, timings = [], []
 for file in os.listdir('timings'):
     with open(f'timings/{file}', 'r') as f:
         n, t = f.read().strip().split(',')
-        threads.append(int(n))
-        timings.append(float(t))
+        if n != "None":
+            threads.append(int(n))
+            timings.append(float(t))
 threads = np.array(threads)
 timings = np.array(timings)
 
@@ -17,7 +19,13 @@ print('This is the data I loaded: threads =', threads, ', timings =',timings)
 fig, axs = plt.subplots()
 
 # CREATE YOUR PLOT HERE
+plt.plot(threads, timings, ".")
+plt.xlabel("# of Threads")
+plt.ylabel("Timings")
+plt.title('Time vs Thread')
+
 # Remember to label your axis
 # Feel free to make it pretty
 
 plt.savefig('threads_v_timings.png', dpi=300)
+plt.show()