Numeri/plots.py

import pandas as pd
import numpy as np

GRADE_LIST = "Cijfers-HerrewijnenJonathan.csv"

class Cijferlijst:
    def __init__(self, grade_list=GRADE_LIST):
        self.grade_list = grade_list
        self.ps = pd.read_csv(GRADE_LIST, skiprows=2, sep=';')

    '''
    Get all data from a subject and a (yearly) period:
    GetSubjectByYear("godsdienst", "2010/2011")
    '''
    def GetSubjectByYear(self, subject, year):
        if(type(subject) == str and type(year) == str):
            select_year = self.ps[self.ps['Schooljaar(Voortgangsdossier)'] == year]
            return select_year[select_year["Vak(Voortgangsdossier)"] == subject]
        else:
            return False
    
    def iets(self):
        print(self.grade_list)
    
    def CalculateMedian(self, subject, year):
        # Calculates average for all scores without weight!
        Grades = self.ps['Cijfer(Voortgangsdossier)'].loc[:]
        print("blub")