From 12ce8a5e220ad4a59beec2a56c360550c04ab13a Mon Sep 17 00:00:00 2001 From: Kyle Belanger Date: Tue, 20 Aug 2024 08:57:44 -0400 Subject: [PATCH] update to graph --- LearnJulia/src/amtrak.jl | 11 ++++++++--- 1 file changed, 8 insertions(+), 3 deletions(-) diff --git a/LearnJulia/src/amtrak.jl b/LearnJulia/src/amtrak.jl index 6a7034a..904764b 100644 --- a/LearnJulia/src/amtrak.jl +++ b/LearnJulia/src/amtrak.jl @@ -74,11 +74,16 @@ gd = @chain mod_df begin :train_code = levelcode(:train) end end - +colors = cgrad(:Dark2_6) f = Figure(); ax = Axis(f[1,1], xlabel = "Station", ylabel = "Mean Delay (mins)", title = "Mean Delay by Station", xticks = (1:length(levels(gd.station_code)), levels(gd.station)), xticklabelrotation = pi/2) -barplot!(ax, gd.station_code, gd.mean, dodge = gd.train_code, color = gd.train_code) +barplot!(ax, gd.station_code, gd.mean, dodge = gd.train_code, color = colors[gd.train_code]) + +labels = ["$i" for i in unique(gd.train)] +elements = [PolyElement(polycolor = colors[i]) for i in unique(gd.train_code)] + +Legend(f[1,2],elements, labels, "Train Number") + f -levels(gd.station_code) \ No newline at end of file