low_memory=False)
colorindex = 0
patches = list()
for columnName in aggregatedResults.columns.values:
plt.plot(np.array(aggregatedResults[columnName].values).astype(float),
color=colors[colorindex])
patches.append(patch.Patch(color=colors[colorindex], label=columnName))
colorindex += 1
plt.title(ScenarioTitle)
plt.legend(handles=patches)
plt.gca().yaxis.grid(True, which='major', ls='dotted')
plt.show()
if __name__ == "__main__":
fileHandler = FileHandler.CsvTrafficDataHandler(inputDirectory, allColumns,
detectorIds)
preprocessor = TrafficPreprocessor.TrafficPreprocessor(mdHandler='linear')
if shouldImport:
result = fileHandler.ReadFiles()
dataset = fileHandler.dataset
if result:
dataset = preprocessor.ExtractDataFromColumns(
dataset, selectedColumns)
fileHandler.WriteFile(dataset)
else:
raise IOError('Error importing from: ' + inputDirectory)
else:
dataset = pd.read_csv(fileHandler.outputFilePath,
sep=";",
decimal=",",
encoding="utf-8",
