Introductory-Python-Course/Code7.py at main · Janderion47/Introductory-Python-Course · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
def cm(datain): # Calculate Mean
    suma = 0
    count=int(0)
    for i in datain:
        suma+=i
        count+=1

    return suma/count

def linreg(points): # Expects each point as list of [x,y]
    x_items = []
    y_items = []
    for i in points:
        x_items.append(i[0])
        y_items.append(i[1])

    x_mean = cm(x_items)
    y_mean = cm(y_items)

    numer = 0
    denom = 0
    for i in data_points:
        numer+=(i[0]-x_mean)*(i[1]-y_mean)
        denom+=((i[0]-x_mean)**2)

    slope = numer/denom
    yint = y_mean - (slope*x_mean)

    return slope, yint

def make_prediction(xdata, m, yint):
    return (xdata*m)+yint

data_points = [
    (2000, 72.5),
    (2001, 73.1),
    (2002, 73.8),
    (2003, 74.2),
    (2004, 74.7),
    (2005, 75.3),
    (2006, 75.9),
    (2007, 76.5),
    (2008, 76.9),
    (2009, 77.4),
    (2010, 78.0),
    (2011, 78.5),
    (2012, 79.0),
    (2013, 79.5),
    (2014, 80.0),
    (2015, 80.5),
    (2016, 81.0),
    (2017, 81.5),
    (2018, 82.0),
    (2019, 82.5)]

linregdata = linreg(data_points)
print(linregdata)
lr_slope = linregdata[0]
lr_yint = linregdata[1]


keep_going=True
while keep_going:
    try:
        newdata = float(input("New 'x' value to predict the 'y'? Enter nothing to stop the program. "))
        print(make_prediction(newdata,lr_slope,lr_yint))
    except ValueError:
       keep_going=False