Exploration of COVID19 dataset: Effectiveness of Measures against COVID19

import pandas as pd
import matplotlib.pyplot as plt
ASEAN_countries_list = ['Brunei', 'Cambodia', 'Indonesia', 'Laos', 'Malaysia', 'Burma', 'Philippines', 'Singapore', 'Vietnam']

link = 'https://raw.githubusercontent.com/CSSEGISandData/COVID-19/master/csse_covid_19_data/csse_covid_19_time_series/time_series_covid19_confirmed_global.csv'
covid19_data = pd.read_csv(link)
asean_df = covid19_data[covid19_data['Country/Region'].isin(ASEAN_countries_list)]
asean_df.set_index('Country/Region', inplace = True)
asean_df_dropped = asean_df.drop(columns = ['Province/State', 'Lat', 'Long'])
asean_t = asean_df_dropped.T

def country_diff(df, country):
    df_s = df[country].copy()
    df_s_F = df_s.shift(1)
    country_df = pd.concat([df_s, df_s_F], axis=1)
    country_df.columns = [f'{country}_Cases_O', f'{country}_Cases_ShiftF']
    country_df[f'{country}_difference'] = country_df[f'{country}_Cases_O'] - country_df[f'{country}_Cases_ShiftF']
    return country_df

asean_daily_df = pd.DataFrame()
for country in ASEAN_countries_list:
    temp_df = country_diff(asean_t, country)
    asean_daily_df = pd.concat([asean_daily_df, temp_df[f'{country}_difference']],sort = False, axis = 1)
asean_daily_df = asean_daily_df.iloc[1:,]
asean_daily_df

asean_daily_df.index = pd.to_datetime(asean_daily_df.index)
asean_daily_df

fig, ax = plt.subplots(figsize = (15,10))

asean_daily_df['Singapore_difference'].plot(ax=ax)
ax.set_ylabel('Confirmed Daily Cases')
ax.set_xlabel('Dates')
ax.legend()
plt.show()

from datetime import date

circuit_breaker_date = date(2020,4,7)
cb_2_date = date(2020,4 ,21)
phase_1_date = date(2020, 6, 2)
phase_2_date = date(2020, 6, 19)
phase_3_date = date(2020, 12, 28)
phase_3_HA_date =  date(2021, 5, 8)
phase_2_HA_date = date(2021, 5, 16)
phase_3_HA2_date = date(2021, 6, 14)
phase_3_5_date = date(2021, 7, 12)
phase_2_HA2_date = date(2021, 7, 22)


#We are going to use a dictionary here to identify each date's details!
dates_dist = {circuit_breaker_date:['Circuit Breaker', 'Tighten'],
              cb_2_date:['Circuit Breaker 2', 'Tighten'],
              phase_1_date:['Phase 1', 'Relax'],
              phase_2_date:['Phase 2', 'Relax'],
              phase_3_date:['Phase 3', 'Relax'],
              phase_3_HA_date:['Phase 3 HA', 'Tighten'],
              phase_2_HA_date:['Phase 2 HA', 'Tighten'],
              phase_3_HA2_date:['Phase 3 HA2', 'Relax'],
              phase_2_HA2_date:['Phase 2 HA2', 'Tighten'],
              phase_3_5_date:['Group of 5 Allowed', 'Relax']}

col_dict = {'Tighten': 'r', 'Relax':'g'}

fig, ax = plt.subplots(figsize = (15,10))
asean_daily_df['Singapore_difference'].plot(ax=ax, label = '')

#Iterate over the keys of the dictionary. aka the different dates
for dates in dates_dist:
    #plot the date, for colour, use the dictionary to see if "Tighten" or "Relax"
    #Then use that as a key for our col_dict to specify our colour (red / green)
    ax.axvline(dates, color=col_dict[dates_dist[dates][1]], 
               linestyle='--', lw=2)

ax.axvline(circuit_breaker_date,color = 'r', label ='Tighten',linestyle='--', lw=2)
ax.axvline(phase_3_date,color = 'g', label ='Relax',linestyle='--', lw=2)
ax.set_ylabel('Confirmed Daily Cases')
ax.set_xlabel('Dates')
ax.legend()
plt.show()

fig, ax = plt.subplots(figsize = (15,10), nrows = 2)
asean_daily_df['Singapore_difference'].plot(ax=ax[0], label = '')
asean_daily_df['Singapore_difference'].plot(ax=ax[1], label = '')

#Iterate over the keys of the dictionary. aka the different dates
for dates in dates_dist:
    #plot the date, for colour, use the dictionary to see if "Tighten" or "Relax"
    #Then use that as a key for our col_dict to specify our colour (red / green)
    ax[0].axvline(dates, color=col_dict[dates_dist[dates][1]], 
               linestyle='--', lw=2)
    ax[1].axvline(dates, color=col_dict[dates_dist[dates][1]], 
               linestyle='--', lw=2)
    

ax[0].axvline(circuit_breaker_date,color = 'r', label ='Tighten',linestyle='--', lw=2)
ax[0].axvline(phase_3_date,color = 'g', label ='Relax',linestyle='--', lw=2)
ax[0].set_ylabel('Confirmed Daily Cases')
ax[0].set_xlabel('Dates')
ax[0].set_xlim(date(2020, 1, 22), date(2020, 12, 31))
ax[0].legend()
ax[0].set_title('Daily Cases in Singapore for 2020')


ax[1].axvline(circuit_breaker_date,color = 'r', label ='Tighten',linestyle='--', lw=2)
ax[1].axvline(phase_3_date,color = 'g', label ='Relax',linestyle='--', lw=2)
ax[1].set_ylabel('Confirmed Daily Cases')
ax[1].set_xlabel('Dates')
ax[1].set_xlim(date(2021, 1, 1), date(2021, 8, 4))
ax[1].set_ylim(-1, 400)
ax[1].legend()
ax[1].set_title('Daily Cases in Singapore for 2021')

plt.tight_layout()
plt.show()