import matplotlib.pyplot as plt
import matplotlib.cbook as cbook

import numpy as np
import pandas as pd

f = open('elongation.csv', 'r')
x = []
y_average = []
y_extreme = []
for l in f.readlines():
    l = l.split(',')
    x.append(l[0])
    y_average.append(float(l[1]))
    y_extreme.append(float(l[2]))

plt.plot(x, y_average, label = 'average')
plt.plot(x, y_extreme, label = 'extreme')

for i in range(len(x)):
        plt.text(i,y_average[i],f'{y_average[i]:.2f}', color="blue")
for i in range(len(x)):
        plt.text(i,y_extreme[i],f'{y_extreme[i]:.2f}', color="orange")

plt.title('Speedup after optimization')
plt.xlabel('N')
plt.ylabel('elongation')
plt.grid()
plt.legend()
plt.show()