Consolidate code from notebooks to utils.py and constants.py modules.

Author: ray310

src/constants.py

@@ -0,0 +1,26 @@
+""" Project constants, parameters, and settings."""
+
+DAY_OF_WEEK_MAP = {
+    0: "Monday",
+    1: "Tuesday",
+    2: "Wednesday",
+    3: "Thursday",
+    4: "Friday",
+    5: "Saturday",
+    6: "Sunday",
+}
+
+MONTHS_MAP = {
+    1: "Jan",
+    2: "Feb",
+    3: "Mar",
+    4: "Apr",
+    5: "May",
+    6: "Jun",
+    7: "Jul",
+    8: "Aug",
+    9: "Sep",
+    10: "Oct",
+    11: "Nov",
+    12: "Dec",
+}

src/utils.py

@@ -0,0 +1,33 @@
+"""Project helper functions."""
+
+import pandas as pd
+from .constants import DAY_OF_WEEK_MAP
+
+
+def make_week_crosstab(
+    df, divisor, day_of_week_map=DAY_OF_WEEK_MAP.copy(), values=None, aggfunc=None
+):
+    """Returns an hour / day-of-week crosstab scaled by a divisor."""
+    ct = pd.crosstab(
+        index=df["datetime"].dt.dayofweek,
+        columns=df["datetime"].dt.hour,
+        values=values,
+        aggfunc=aggfunc,
+    )
+    if day_of_week_map:
+        ct.rename(index=day_of_week_map)
+    ct /= divisor  # scale crosstab by divisor
+    return ct
+
+
+def make_heatmap_labels(
+    title, x_label="Hour of Day", y_label="", cbar_label="Number of Collisions per Hour"
+):
+    """Returns a dictionary of labels for a 2D heatmap."""
+    ct_labels = {
+        "title": title,
+        "x_label": x_label,
+        "y_label": y_label,
+        "cbar_label": cbar_label,
+    }
+    return ct_labels