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",