公交车到站预测1----python处理csv文件

# -*- coding: utf-8 -*-
"""
Created on Thu Jan 14 21:41:20 2016

@author: Luyixiao
"""
import csv#import the csv module

def csv2dict(path):
    reader = csv.reader(open(path,'rb'))
    csvList = []
    startNoList=[]
    stationKey = {}
    sectList = []
    li =[]
    tempKey = {}
    for line in reader:#change the csv file to a list
        csvList.append(line)

    for i in range(0,len(csvList)):#get the bit of the start record
        if csvList[i][0][0]=='a':
            startNoList.append(i)               
   
    for i in range(0,len(csvList)):#tranverse the csv list ,like going through the file
        if i in startNoList:
            if i == 0:#if this is a begining of the first record
                key = csvList[i][0]#mark the key 
            else:#if this is a begining of a record
                stationKey[key] = sectList#put the value to the key,the value is a list of the next data
                sectList = []#empty the sectList,which store the csv lines temporaryly
                key = csvList[i][0]
        if not i in startNoList:#this is the content of the record
            sectList.append(csvList[i])
   
    for k in stationKey:#tranverse the dict which store all the content as a record
       li = stationKey[k]#get a key's value
       for i in range(0,len(li)):
           tempKey[li[i][0]]=li[i][1:3]#create a dict as the value of this key
       stationKey[k]=tempKey
    return stationKey