COVID-19 analysis code 04 - plot cases

covid19_analysis_04_plot_cases.py

# Estimate a reasonable value for Y tick given the maximum number of cases
Y_GRID_TICK = 10**int(np.log10(country_df['Confirmed'].max()))/2

ax = plt.gca()

sns.set_style("whitegrid", {'grid.linestyle': ':'})
ax.xaxis.set_major_locator(ticker.MultipleLocator(4))

# Plot 
country_df.plot(x='Date', y=["Confirmed", "Deaths", "Recovered"], figsize=(18,9), ax=ax, marker='o')
country_df.plot(x='Date', y=["Increase"], secondary_y=True, figsize=(18,9), ax=ax, marker='o')

# Improve display of 
ax.xaxis.set_major_formatter(mdates.DateFormatter('%d-%m-%Y'))

# Set y label and plot title
ax.set_ylabel("# of confirmed cases");
ax.set_title("Daily numbers of confirmed cases, deaths and hospitalized people. Daily increase refers to numbers on the right y axis.");

# Uncomment if you want to save the image
# plt.savefig("daily_data.png");