python-class/readrides.py at master · sunithak/python-class · 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
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
# readrides.py

import csv

def read_rides_as_tuples(filename):
    '''
    Read the bus ride data as a list of tuples
    '''
    rides = []
    f = open(filename)
    f_csv = csv.reader(f)
    headings = next(f_csv)     # Skip headers
    for row in f_csv:
        route = row[0]
        date = row[1]
        daytype = row[2]
        numrides = int(row[3])
        ride = (route, date, daytype, numrides)
        rides.append(ride)
    f.close()
    return rides

def read_rides_as_dicts(filename):
    '''
    Read the bus ride data as a list of dicts
    '''
    rides = []
    f = open(filename)
    f_csv = csv.reader(f)
    headings = next(f_csv)     # Skip headers
    for row in f_csv:
        route = row[0]
        datefields = row[1].split('/')
        daytype = row[2]
        numrides = int(row[3])
        ride = {
            'route': route,
            'year': int(datefields[2]),
            'month': int(datefields[0]),
            'day': int(datefields[1]),
            'daytype': daytype,
            'numrides': numrides
        }
        rides.append(ride)
    f.close()
    return rides

class Ride(object):
    # Uncomment to see effect of slots
    # __slots__ = ('route', 'date', 'daytype', 'numrides')
    def __init__(self, route, date, daytype, numrides):
        self.route = route
        self.date = date
        self.daytype = daytype
        self.numrides = numrides

# Uncomment to use a namedtuple instead
#from collections import namedtuple
#Ride = namedtuple('Ride',('route','date','daytype','numrides'))

def read_rides_as_instances(filename):
    '''
    Read the bus ride data as a list of instances
    '''
    rides = []
    f = open(filename)
    f_csv = csv.reader(f)
    headings = next(f_csv)     # Skip headers
    for row in f_csv:
        route = row[0]
        date = row[1]
        daytype = row[2]
        numrides = int(row[3])
        ride = Ride(route, date, daytype, numrides)
        rides.append(ride)
    f.close()
    return rides

if __name__ == '__main__':
    import time
    import os

    read_rides = read_rides_as_tuples # Change to as_dicts, as_instances, etc.
    start = time.time()
    rides = read_rides("Data/ctabus.csv")
    end = time.time()
    print("Took", end-start, "seconds")
    print("Look at memory use of PID", os.getpid())