Alvianojs commited on
Commit
667e07a
·
1 Parent(s): 0d47fe9

Update src/map_viz_pred.py

Browse files
Files changed (1) hide show
  1. src/map_viz_pred.py +5 -5
src/map_viz_pred.py CHANGED
@@ -52,10 +52,10 @@ def calling_pred_map_viz(counts_df1):
52
  count_min = day_hour_view_group['total'].min()
53
 
54
 
55
- count_dict = {"woodlands_to_sg" :date_view_group.loc[(date_view_group['view'] == 'Woodlands - to SG') & (date_view_group['day_name'] == filtered_day) & (date_view_group['hour'] == filtered_hour), "total" ].iloc[0],
56
- "woodlands_to_jh" :date_view_group.loc[(date_view_group['view'] == 'Woodlands - to Johor') & (date_view_group['day_name'] == filtered_day) & (date_view_group['hour'] == filtered_hour), "total" ].iloc[0],
57
- "tuas_to_sg" :date_view_group.loc[(date_view_group['view'] == 'Tuas - to SG') & (date_view_group['day_name'] == filtered_day) & (date_view_group['hour'] == filtered_hour), "total" ].iloc[0],
58
- "tuas_to_jh" :date_view_group.loc[(date_view_group['view'] == 'Tuas - to Johor') & (date_view_group['day_name'] == filtered_day) & (date_view_group['hour'] == filtered_hour), "total" ].iloc[0]
59
  }
60
 
61
  values = np.array([0., 0.5, 1.])
@@ -92,5 +92,5 @@ def calling_pred_map_viz(counts_df1):
92
  ax.add_artist(collection)
93
  ax.set_xlim([100, 1900])
94
  ax.set_ylim([1800, 0])
95
- ax.set_title(datetime.datetime.strptime(filtered_day + " | " + filtered_hour + " SGT", fontname = 'Georgia')
96
  return fig
 
52
  count_min = day_hour_view_group['total'].min()
53
 
54
 
55
+ count_dict = {"woodlands_to_sg" :day_hour_view_group.loc[(day_hour_view_group['view'] == 'Woodlands - to SG') & (day_hour_view_group['day_name'] == filtered_day) & (day_hour_view_group['hour'] == filtered_hour), "total" ].iloc[0],
56
+ "woodlands_to_jh" :day_hour_view_group.loc[(day_hour_view_group['view'] == 'Woodlands - to Johor') & (day_hour_view_group['day_name'] == filtered_day) & (day_hour_view_group['hour'] == filtered_hour), "total" ].iloc[0],
57
+ "tuas_to_sg" :day_hour_view_group.loc[(day_hour_view_group['view'] == 'Tuas - to SG') & (day_hour_view_group['day_name'] == filtered_day) & (day_hour_view_group['hour'] == filtered_hour), "total" ].iloc[0],
58
+ "tuas_to_jh" :day_hour_view_group.loc[(day_hour_view_group['view'] == 'Tuas - to Johor') & (day_hour_view_group['day_name'] == filtered_day) & (day_hour_view_group['hour'] == filtered_hour), "total" ].iloc[0]
59
  }
60
 
61
  values = np.array([0., 0.5, 1.])
 
92
  ax.add_artist(collection)
93
  ax.set_xlim([100, 1900])
94
  ax.set_ylim([1800, 0])
95
+ ax.set_title(filtered_day+ " | " + filtered_hour + " SGT", fontname = 'Georgia')
96
  return fig