diff --git a/Analyze.py b/Analyze.py index 3f07dbf..a4ce6d2 100644 --- a/Analyze.py +++ b/Analyze.py @@ -14,7 +14,7 @@ def compute_avg_delay(data): """ Compute the average delay for each route """ - return data.groupby('Route').mean().reset_index() + return data.groupby('Route')['Min Delay'].mean().reset_index() def main():