time_series_plots.py

import matplotlib.pyplot as plt
from rate_calculations import time_series

diseases = ["cvd", "respiratory_disease", "cancer"]

for pop in ['total', 'emergency', 'elective']:
    

    fig, axes = plt.subplots(ncols=1, nrows=3, sharey=False, figsize=[10, 12])
    for i, ax in enumerate(axes.flat):
        df = time_series[pop][i]
        df.plot(ax=ax, color=["#176dde", "#ffad33"])
        ax.grid(which="both", axis="y", color="#666666", linestyle="-", alpha=0.2)
    #     name = measures[i].numerator.replace("_", " ")
        title = f"{chr(97 + i)}) {diseases[i]} hospitalisation rates"
        ax.set_title(title, loc="left")
        ax.xaxis.label.set_visible(False)
        ax.legend(loc=3, prop={"size": 9})
        ax.set_ylabel("Rate per 100,000 people")
        ax.set_ylim(ymin=0)
        plt.tight_layout()
    plt.savefig(f"output/{pop}_time_series_plot.svg")