import numpy as np
import matplotlib.pyplot as plt

M = np.loadtxt('ra_data.txt', delimiter=' ')

Kj = M[:, 0]  # kjonn, men brukes ikke i denne oppgaven
Kl = M[:, 1]  # klasse 
Timer = M[:, 2]  # antall timer skjermbruk per pers

t8 = 0
t9 = 0
t10 = 0

for n in range(0, len(Timer), 1):
    
    if (Kl[n] == 8):
        t8 = t8 + Timer[n]
        
    elif(Kl[n] == 9):
        t9 = t9 + Timer[n]
        
    else:    
        t10 = t10 + Timer[n]

klassetrinn = ['8. klasse', '9. klasse', '10. klasse']

plt.close('all')
plt.bar(klassetrinn, [t8, t9, t10])
plt.title('Total skjermbruk per uke per klassetrinn')
plt.ylabel('[timer]')

plt.savefig('plott_ra_stolpe.pdf')
plt.show()