# Starting from 0, actual effort is indexed at 17
# Starting from 0, team size is indexed at 43
# Starting from 0, Programmer experience rated from scale of 1 to 5 indexed at 39 
import numpy
from matplotlib import pyplot as pl
from sklearn.linear_model import LinearRegression
import pandas
dat = pandas.read_csv("dat.csv") #Read data
e = dat["Actual effort"] #Actual effort (cost)
print(e)
i = 0
resources = {}
for size in dat["Team size"].tolist():
    resources[str(i)] = size * 1/dat["Programmers capability "][i]
    i=i+1

s =  #Team size and programmer's capability (resources)
lr = LinearRegression()
lr.fit(s, e)
lobf = lr.predict(s)
pl.scatter(s, e)
pl.plot(s, lobf, color="red")
pl.show()