Source code for cowidev.hosp.sources.serbia
import pandas as pd
from cowidev.utils.clean import clean_date_series
METADATA = {
"source_url": "https://raw.githubusercontent.com/aleksandar-jovicic/COVID19-Serbia/master/timeseries.csv",
"source_url_ref": "https://github.com/aleksandar-jovicic/COVID19-Serbia",
"source_name": "Ministry of Health via github.com/aleksandar-jovicic/COVID19-Serbia",
"entity": "Serbia",
}
[docs]def main():
df = (
pd.read_csv(METADATA["source_url"], usecols=["ts", "hospitalized", "ventilated"])
.rename(
columns={"ts": "date", "hospitalized": "Daily hospital occupancy", "ventilated": "Daily ICU occupancy"}
)
.melt(id_vars="date", var_name="indicator", value_name="value")
.assign(
entity=METADATA["entity"],
)
)
df["date"] = clean_date_series(df.date, "%Y-%m-%d %H:%M:%S")
df = df.drop_duplicates(subset=["date", "indicator"])
return df, METADATA
if __name__ == "__main__":
main()