xox/coronavis
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

all basic plots

Browse Source
This commit is contained in:
fordprefect
2020-03-20 14:42:43 +01:00
parent 118214ba11
commit 1677f19372
1 changed files with 15 additions and 17 deletions
Download Patch File Download Diff File Expand all files Collapse all files

32
basics.py
View File

@@ -5,34 +5,32 @@ import matplotlib.pyplot as pp
def plot(data, countries): def plot(data, countries):
tcp, tc = pp.subplots()
tdp, td = pp.subplots()
ncp, nc = pp.subplots()
ndp, nd = pp.subplots()
for loc in data: for loc in data:
if loc not in countries: if loc not in countries:
continue continue
time, new_cases, new_deaths, total_cases, total_deaths = data[loc] time, new_cases, new_deaths, total_cases, total_deaths = data[loc]
# total cases # total cases
pp.figure("total_cases") tc.plot(time, total_cases, label=f"{loc}")
pp.plot(time, total_cases, label=f"{loc}")
# total deaths # total deaths
pp.figure("total_deaths") td.plot(time, total_deaths, label=f"{loc}")
pp.plot(time, total_deaths, label=f"{loc}")
# new cases # new cases
pp.figure("new_cases") nc.plot(time, new_cases, label=f"{loc}")
pp.plot(time, new_cases, label=f"{loc}")
# new deaths # new deaths
pp.figure("new_deaths") nd.plot(time, new_deaths, label=f"{loc}")
pp.plot(time, new_deaths, label=f"{loc}")
for name in ["total_cases", "total_deaths", "new_cases", "new_deaths"]: for ax, fig, name in [(tc, tcp, "total_cases"), (td, tdp, "total_deaths"), (nc, ncp, "new_cases"), (nd, ndp, "new_deaths")]:
postprocess(name) ax.set_yscale("log")
for tick in ax.get_xticklabels():
tick.set_rotation(45)
ax.legend(frameon=False)
fig.tight_layout()
def postprocess(name): fig.savefig(name+".png")
pp.yscale("log")
pp.xticks(rotation=90)
pp.legend(frameon=False)
pp.tight_layout()
pp.savefig(f"{name}.png")