import matplotlib.pyplot as plt, numpy as np

l0 = 50.8e-3
d = 12.8e-3

F = [0,7.33,15.1,23.1,30.4,34.4,38.4,41.3,44.8,46.2,47.3,47.5,46.1,44.8,42.6,36.4]
l = [50.8,50.851,50.902,50.952,51.003,51.054,51.308,51.816,52.832,53.848,54.864,55.88,56.896,57.658,58.42,59.182]

F = np.array(F)
l = np.array(l)
l *= 10**(-3)
F *= 10**3

A0 = np.pi/4 * (d**2)

sigma = F/A0
delta_l = l-l0
eps = delta_l/l0

element_nr = [i for i in range(len(sigma))]
x_akse = np.linspace(0.02,0.03, 3)

elastisitetsmodulus = 58.5e9
toyning_offset = elastisitetsmodulus * x_akse - 0.02*elastisitetsmodulus

print(max(sigma)) #finner strekkfastheten
print((delta_l[-1]/l0)*100) #finner duktiliteten

i = 5
plt.plot(element_nr[:i],sigma[:i]/eps[:i]) #finner elastisitetsmodulen
plt.savefig('elastisitetsmodul.png')
plt.show()
plt.plot(eps, sigma)
plt.plot(x_akse,toyning_offset)
plt.savefig('figur.png')
plt.show()