import matplotlib.pyplot as plt
import matplotlib as mpl

matlab = (0.940, 1.806, 2.678, 3.564, 4.411, 5.264)
mf = (0.055, 0.110, 0.162, 0.212, 0.266, 0.320)

matlab_ms = []
mf_ms = []

for time in matlab:
	matlab_ms.append(time*1000)

for time in mf:
	mf_ms.append(time*1000)

fig = plt.figure(figsize=(4,4), facecolor='w', edgecolor='k') 
ax = fig.add_subplot(111)	
ax.plot((1,2,3,4,5,6), matlab, 'b', linewidth=2, marker='o',
        markerfacecolor='b', markersize=7)
ax.plot((1,2,3,4,5,6), mf,'r', linewidth=2, linestyle='dashed', marker='x',
        markerfacecolor='r', markersize=7, markeredgewidth=2)

for tick in plt.gca().xaxis.get_major_ticks(): tick.label1.set_fontsize(10) 
for tick in plt.gca().yaxis.get_major_ticks(): tick.label1.set_fontsize(10)
mpl.rcParams['axes.linewidth'] = 0.8 
ax.set_xlabel('Number of experiment sweeps')
ax.set_ylabel('CPU time (s)')
#ax.yaxis.grid(color='gray', linestyle='dashed')
fig.tight_layout()
plt.xlim(.5, 6.5)
plt.legend(('Matlab', 'ModFossa'), loc='upper left', fontsize=10, numpoints=1)

plt.show()
fig.savefig('runtime.eps', format='eps')