Creating Heatmap 1

heatmap1.py

df_10 = df[df['STATION'].isin(list(group_station.head(10).index))]
df_10['WEEKDAY'] = df_10['DATE'].dt.day_name()
group_station_day = df_10.groupby(['STATION','WEEKDAY'])['ENTRY_EXIT'].sum()
matrix_station_day = group_station_day.unstack()
matrix_station_day.reset_index()
matrix_station_day = matrix_station_day.reindex(columns=["Monday","Tuesday","Wednesday","Thursday","Friday","Saturday","Sunday"])
matrix_station_day = matrix_station_day.reindex(index=list(group_station.head(10).index))

array = np.array(matrix_station_day.applymap(lambda x:str(round(x/1000,1))+'k'))
fig2 = plt.figure(figsize=[10,10])
cmap = sns.cubehelix_palette(light=1, as_cmap=True)
ax2 = sns.heatmap(matrix_station_day,cmap='Blues',linecolor='white',linewidths=1,annot = array,fmt='')
plt.xlabel('Day of the Week',fontsize=15)
plt.ylabel('Top 10 Stations',fontsize=15)   
plt.title('Station Traffic in the Week',weight='bold',fontsize=15)