sbrandeis HF staff commited on
Commit
e0bcfa9
1 Parent(s): 4c0a5e2

Add titleé

Browse files
Files changed (1) hide show
  1. 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