import numpy as np
import matplotlib.pyplot as plt


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

Kj = M[:, 0] #  kjoenn
Kl = M[:, 1] #  klassetrinn
Timer = M[:, 2] #  antall timer skjermbruk per pers

t8g = 0
t8j = 0
t9g = 0
t9j = 0
t10g = 0
t10j = 0

for n in range(0, len(Kj), 1):
    
    if (Kl[n] == 8 and Kj[n] == 1):
        t8g = t8g + Timer[n]
        
    elif(Kl[n] == 8 and Kj[n] == 2):
        t8j = t8j + Timer[n]
        
    elif (Kl[n] == 9 and Kj[n] == 1):
        t9g = t9g + Timer[n]
        
    elif(Kl[n] == 9 and Kj[n] == 2):
        t9j = t9j + Timer[n] 
        
    elif (Kl[n] == 10 and Kj[n] == 1):
        t10g = t10g + Timer[n]
        
    else:
        t10j = t10j + Timer[n]    
        

labels = ['8.Klg', '8.Klj', '9.Klg', '9.Klj',
        '10.Klg', '10.Klj']
sizes = [t8g, t8j, t9g, t9j, t10g, t10j]

plt.close('all')

plt.bar(labels, sizes)
plt.ylabel('[Timer]')
plt.xlabel('j = jenter. g = gutter.')

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