-
Notifications
You must be signed in to change notification settings - Fork 1
/
schooldata.py
54 lines (45 loc) · 1.43 KB
/
schooldata.py
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
#f.close()
import csv
import random
class SchoolData():
def __init__(self):
print "creating SchoolData obj"
def get_data_from_csv(self, priority): #pass in a priority (1-3) based on how much they value school
f = open('schools.csv', 'rb')
reader = csv.reader(f)
zip_codes=[]
if priority==1:
for items in reader:
try:
if float(items[16]) > 65:
zip_codes.append(items[7])
except:
pass
return zip_codes
if priority==2:
for items in reader:
try:
if float(items[16]) < 65 and float(items[16]) > 45:
zip_codes.append(items[7])
except:
print items[16]
return zip_codes
if priority==3:
for items in reader:
try:
if float(items[16]) < 45:
zip_codes.append(items[7])
except:
pass
return zip_codes
def get_zip(self, priority):
data = self.get_data_from_csv(priority)
print data
data = random.choice(data)
try:
data = data[-10:]
data = data[:5]
except:
self.get_zips(priority)
return data
#jprint SchoolData().get_zip(1)