calc.py

from json.tool import main
import pandas as pd
import sys
def calc_loads():
    data = pd.read_csv("data.csv")
    # print(data)
    loads = []
    for index,i in data.iterrows():
        if i["gender"] == "M":
            load = i["age"]*0.5+i['severity']*2+i['weight']*0.5
        elif i["gender"] == "F":
            load = i["age"]*0.5+0.5*0.2+i['severity']*2+i['weight']*0.5
        loads.append(load)
    
    return loads


# each floor has 20 rooms and each floor has 2 sides and two exits in middle
def get_floor_weights(loads):
    weight_floor = []
    for i in range(0,len(loads),20):
        sum_a = 0
        sum_b = 0
        for a in range(i,i+10):
            sum_a +=loads[a]
        for b in range(i+10,i+20):
            sum_b +=loads[b]
        weight_floor.append([sum_a,sum_b])
    # print(weight_floor)
    return weight_floor


if __name__ == "__main__":
    # args = sys.argv[1:]
    # print(args[0])
    loads= calc_loads()
    get_floor_weights(loads)