def getPitStops(filepath):
"""
Parameters
----------
filepath: string
The directory to the file relative to the parentPath
# [raceId, driverId, stop, lap, time, duration, milliseconds]
Returns
-------
tuple
(headings, data, dictionary)
"""
RecentDriversList = getRecentDrivers(2000)
circuitRaceList = getCircuitRaceList("races.csv")
circuitRaceKeys = circuitRaceList.keys()
PitStopsData = getDataset(filepath)
dataObtained = PitStopsData.next()
Data = []
for data in PitStopsData:
if (ifExist(data[1], RecentDriversList)):
for key in circuitRaceKeys:
if (data[0] in circuitRaceList[key]):
Data.append(data)
# Data = list(PitStopsData)
print "Data from file: " + str(dataObtained)
# print "Data Extracted: " + 'Year ' + " raceID"
_ret_dict = defaultdict(lambda: defaultdict(lambda: 0.0))
for d in Data:
# {raceId, driverId :{ stop: pit time }}
_ret_dict[str(d[0]) + "," + str(d[1])][d[2]] = d[6]
return (dataObtained, Data, _ret_dict)
def getDataFromResults(filepath):
"""
Parameters
----------
filepath: string
The directory to the file relative to the parentPath
# [raceID, driverID, constructorID, rank, points, grid, postion, positionOrder]
Returns
-------
tuple
(headings, data)
"""
RecentDriversList = getRecentDrivers(2000)
circuitRaceList = getCircuitRaceList("races.csv")
circuitRaceKeys = circuitRaceList.keys()
ResultData = getDataset(filepath)
dataObtained = ResultData.next()
Data = []
# Data = list(ResultData)
for data in ResultData:
if (ifExist(data[1], RecentDriversList)):
for key in circuitRaceKeys:
if (data[0] in circuitRaceList[key]):
Data.append(data)
print "Data from file: " + str(dataObtained)
# print "Data Extracted: " + 'Year ' + " raceID"
# yearDict_raceList = defaultdict(lambda:[]) # initialise the dictionary value to an empty list
# dataObtained.append("firstLapChange")
# for d in Data:
# d.append(int(d[5]) - int(d[7]))
return (dataObtained, Data)
def getPreprocessedData(filepath):
"""
Parameters
----------
filepath: string
The directory to the file relative to the parentPath
# [raceId, driverId, constructorId, rank, points, grid, position, positionOrder, positionText , firstLapChange, pitStops]
Returns
-------
tuple
(headings, data, dictionary)
"""
RecentDriversList = getRecentDrivers(2000)
circuitRaceList = getCircuitRaceList("races.csv")
circuitRaceKeys = circuitRaceList.keys()
PreprocessedData = getDataset(filepath)
dataObtained = PreprocessedData.next()
# Data = list(PreprocessedData)
Data = []
for data in PreprocessedData:
if (ifExist(data[1], RecentDriversList)):
for key in circuitRaceKeys:
if (data[0] in circuitRaceList[key]):
Data.append(data)
print "Data from file: " + str(dataObtained)
return (dataObtained, Data)
def getRecentTracks():
"""
Returns:
--------
list:
[CircuitIDs]
"""
TrackHeadings, TrackData = getDataFromTracks("track_data_final.csv")
_ret_list = []
for data in TrackData:
if ifExist(data[1], _ret_list):
i = 1 # dummy variable
else:
_ret_list.append(data[1])
return _ret_list
def getLapTimes(filepath):
"""
Parameters
----------
filepath: string
The directory to the file relative to the parentPath
# [raceId, driverId, lap, position, time, milliseconds]
Returns
-------
tuple
(headings, data, dictionary, dictionary of dictionary)
"""
RecentDriversList = getRecentDrivers(2000)
circuitRaceList = getCircuitRaceList("races.csv")
circuitRaceKeys = circuitRaceList.keys()
LapTimesData = getDataset(filepath)
dataObtained = LapTimesData.next()
# Data = list(LapTimesData)
Data = []
for data in LapTimesData:
if (ifExist(data[1], RecentDriversList)):
for key in circuitRaceKeys:
if (data[0] in circuitRaceList[key]):
Data.append(data)
print "Data from file: " + str(dataObtained)
# {raceID: [[driverID, lap#, laptime]]}
_ret_dict = defaultdict(lambda: [])
# {raceID: {lap#: [driverID, laptime]}}
_ret_dict_dict = defaultdict(lambda: defaultdict(lambda: []))
for d in Data:
# {raceId, driverId :{ lap: laptime]]
# _ret_dict[str(d[0]) +"," + str(d[1])][d[2]] = d[5]
_ret_dict[int(d[0])].append(np.array([int(d[1]),
int(d[2]),
int(d[5])]))
# {raceID: {lap#: [driverID, laptime]}}
_ret_dict_dict[int(d[0])][int(d[2])].append([int(d[1]), int(d[5])])
# keys = sorted(_ret_dict_dict.keys(), lambda _key: int(_key))
keys = _ret_dict_dict.keys()
for key in keys:
_keys = _ret_dict_dict[key].keys()
for _key in _keys:
_ret_dict_dict[key][_key].sort(key=itemgetter(1))
# print _ret_dict_dict[key][_key]
return (dataObtained, Data, _ret_dict, _ret_dict_dict)