Add titleé
Browse files- heatmap.py +2 -1
heatmap.py
CHANGED
@@ -25,7 +25,7 @@ def build_heatmap(recent_songs: List[dict]) -> np.ndarray:
|
|
25 |
played_at = parse(track["played_at"])
|
26 |
weekday = datetime.weekday(played_at)
|
27 |
week_offset = (now - played_at).days // 7
|
28 |
-
heatmap[weekday, -week_offset] +=1
|
29 |
return heatmap
|
30 |
|
31 |
|
@@ -33,5 +33,6 @@ def plot(heatmap: np.ndarray):
|
|
33 |
fig, ax = plt.subplots()
|
34 |
|
35 |
ax.imshow(heatmap, cmap="Greens")
|
|
|
36 |
ax.set_yticklabels(["Monday", "Tuesday", "Wednesday", "Thursday", "Friday", "Saturday", "Sunday"])
|
37 |
return fig, ax
|
|
|
25 |
played_at = parse(track["played_at"])
|
26 |
weekday = datetime.weekday(played_at)
|
27 |
week_offset = (now - played_at).days // 7
|
28 |
+
heatmap[weekday, -(week_offset +1)] +=1
|
29 |
return heatmap
|
30 |
|
31 |
|
|
|
33 |
fig, ax = plt.subplots()
|
34 |
|
35 |
ax.imshow(heatmap, cmap="Greens")
|
36 |
+
ax.set_title("Recent activity")
|
37 |
ax.set_yticklabels(["Monday", "Tuesday", "Wednesday", "Thursday", "Friday", "Saturday", "Sunday"])
|
38 |
return fig, ax
|