Luiz Gustavo Assignment Pull Request

Analyze.py

@@ -10,19 +10,19 @@ def import_data():
     return pd.read_csv('ttc-bus-delay-data-2023.csv')
 
 
-def compute_avg_delay(data):
+def compute_avg_delay(data, group_by: str = 'Route'):
     """
-    Compute the average delay for each route
+    Compute the average delay aggregated by specified column
     """
-    return data.groupby('Route').mean().reset_index()
+    return data.groupby('Route')['Min Delay'].mean().reset_index()
 
 
 def main():
     """
     Main function
     """
     data = import_data()
-    avg_delay = compute_avg_delay(data)
+    avg_delay = compute_avg_delay(data, group_by='Route')
 
     plt.hist(avg_delay['Min Delay'], bins=30)
     plt.title('Average Delay Distribution')