refactor

total_cases.py

@@ -0,0 +1,23 @@
+"""
+Plot total cases of countries over time on log scale
+"""
+import matplotlib.pyplot as pp
+name="total_cases"
+
+def plot(data, countries):
+
+    for loc in data:
+        if loc not in countries:
+            continue
+        time, new_cases, new_deaths, total_cases, total_deaths = data[loc]
+        
+        # total cases
+        pp.figure(name)
+        pp.plot(time, total_cases, label=f"{loc}")
+
+    pp.yscale("log")
+    pp.xticks(rotation=90)
+    pp.legend(frameon=False)
+    pp.tight_layout()
+
+    pp.savefig(name+".png")