# ---
# jupyter:
#   jupytext:
#     text_representation:
#       extension: .py
#       format_name: light
#       format_version: '1.5'
#       jupytext_version: 1.14.4
#   kernelspec:
#     display_name: 'Python 3.7.5 64-bit (''venv'': virtualenv)'
#     name: python37564bitvenvvirtualenv1fc8950713be40ee9a21293a7b19d27b
# ---

from covid_health.prep_ecdc import parse_covid_world_data
import matplotlib.pyplot as plt
import seaborn as sns

df = parse_covid_world_data()
print(df.shape)
df.head()

subset = df.loc[
    df.geo.isin(
        df.groupby("geo")
        .active_cases.max()
        .nlargest(5)
        .index
    )
]
sns.lineplot(data=subset, x="time", y="active_cases", hue="geo")



StarcoderdataJupyterScriptsDedupFiltered