diff --git a/exercises/exerciseA/plot.py b/exercises/exerciseA/plot.py index 4130224..caf9278 100644 --- a/exercises/exerciseA/plot.py +++ b/exercises/exerciseA/plot.py @@ -1,14 +1,19 @@ import os import numpy as np import matplotlib.pyplot as plt +import pdb +import glob # IO: This loads the timings for you threads, timings = [], [] -for file in os.listdir('timings'): +files = os.listdir('timings') +files.sort(key= lambda x: int(x.split('_')[0])) +for file in files: with open(f'timings/{file}', 'r') as f: - n, t = f.read().strip().split(',') - threads.append(int(n)) - timings.append(float(t)) + if 'None' not in file: + n, t = f.read().strip().split(',') + threads.append(int(n)) + timings.append(float(t)) threads = np.array(threads) timings = np.array(timings) @@ -17,7 +22,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('Threads') +plt.ylabel('Timings') +plt.xscale('log') + # Remember to label your axis # Feel free to make it pretty + plt.savefig('threads_v_timings.png', dpi=300) diff --git a/exercises/exerciseA/threads_v_timings.png b/exercises/exerciseA/threads_v_timings.png new file mode 100644 index 0000000..5e0ced0 Binary files /dev/null and b/exercises/exerciseA/threads_v_timings.png differ