Update.
authorFrancois Fleuret <francois@fleuret.org>
Thu, 12 Mar 2020 13:13:34 +0000 (14:13 +0100)
committerFrancois Fleuret <francois@fleuret.org>
Thu, 12 Mar 2020 13:13:34 +0000 (14:13 +0100)
covid19.py

index 0fd9ecc..b377458 100755 (executable)
@@ -44,8 +44,8 @@ with open(file, newline='') as csvfile:
 fig = plt.figure()
 ax = fig.add_subplot(1, 1, 1)
 
-ax.grid(color='gray', linestyle='-', linewidth=0.25)
-
+# ax.grid
+ax.yaxis.grid(color='gray', linestyle='-', linewidth=0.25)
 ax.set_title('Nb. of COVID-19 cases')
 ax.set_xlabel('Date', labelpad = 10)
 ax.set_yscale('log')
@@ -54,12 +54,17 @@ myFmt = mdates.DateFormatter('%b %d')
 ax.xaxis.set_major_formatter(myFmt)
 dates = mdates.epoch2num(times)
 
-for label, color in [ ('World', 'blue'),
-                      ('Switzerland', 'red'),
-                      ('France', 'green'),
-                      ('South Korea', 'gray'),
-                      ('Mainland China', 'orange') ]:
-    ax.plot(dates, nb_cases[label], color = color, label = label)
+for key, color, label, delta in [
+        ('World', 'blue', 'World', 0),
+        ('Switzerland', 'red', 'Switzerland', 14),
+        ('France', 'lightgreen', 'France', 11),
+        ('US', 'black', 'USA', 14),
+        ('Korea, South', 'gray', 'S. Korea', 0),
+        ('Italy', 'purple', 'Italy', 3),
+        ('China', 'orange', 'China', 0)
+]:
+    delta = 0
+    ax.plot(dates[:dates.shape[0]-delta], nb_cases[key][delta:], color = color, label = label, linewidth=2)
 
 # ax.legend(loc='center left', bbox_to_anchor=(1, 0.5), frameon = False)
 ax.legend(frameon = False)