readCSV.py

import csv
import numpy as np

people_to_building_list = []
building_to_station_list = []

with open('./data/people_to_building.csv', 'rb') as csvfile:
    raw = []

    spamreader = csv.reader(csvfile, delimiter=',')
    for row in spamreader:
        nrow = []
        for c in range(1,len(row)):
            nrow.append(row[c])
        raw.append(nrow)
        print nrow

    for r in range(1,len(raw)):
        people_to_building_list.append(raw[r])

with open('./data/building_to_station.csv', 'rb') as csvfile:
    raw = []

    spamreader = csv.reader(csvfile, delimiter=',')
    for row in spamreader:
        nrow = []
        for c in range(1,len(row)):
            nrow.append(row[c])
        raw.append(nrow)
        print nrow

    for r in range(1,len(raw)):
        building_to_station_list.append(raw[r])

people_to_building = np.array(people_to_building_list)
building_to_station = np.transpose(np.array(building_to_station_list))

print people_to_building
print building_to_station