From a9f72a0b70db00b809370cc473f0bc022d67efbd Mon Sep 17 00:00:00 2001 From: fordprefect Date: Thu, 23 Sep 2021 10:22:22 +0200 Subject: [PATCH] plot booster vaccinations --- all_countries.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/all_countries.py b/all_countries.py index 8d42330..8b6be3e 100644 --- a/all_countries.py +++ b/all_countries.py @@ -34,6 +34,7 @@ def plot(data, countries, pop, metadata={}, **kwargs): name = basename+loc time, new_cases, new_deaths, total_cases, total_deaths, total_vaccinations, stringency_index, new_vaccinations = data[loc]['time'], data[loc]['new_cases'], data[loc]['new_deaths'], data[loc]['total_cases'], data[loc]['total_deaths'], data[loc]['total_vaccinations'], data[loc]['stringency_index'], data[loc]['new_vaccinations'] people_fully_vaccinated = data[loc]['people_fully_vaccinated'] + total_boosters = data[loc]['total_boosters'] fig, ax1 = pp.subplots(num=name, figsize=figsize) @@ -56,6 +57,7 @@ def plot(data, countries, pop, metadata={}, **kwargs): ax2.plot(time, np.array(total_vaccinations), label=f"Total vaccination doses", marker="", linestyle="-.", color="crimson") ax2.plot(time, np.array(people_fully_vaccinated), label="fully vaccinated", marker="", linestyle="-", color="crimson") + ax2.plot(time, np.array(total_boosters), label="total booster vaccinations", marker="", linestyle="--", color="crimson") # fix lower bound of plot for ax in (ax1, ax2):