From: Francois Fleuret <francois@fleuret.org>
Date: Mon, 9 Mar 2020 16:07:25 +0000 (+0100)
Subject: Initial commit.
X-Git-Url: https://fleuret.org/cgi-bin/gitweb/gitweb.cgi?a=commitdiff_plain;h=0db243241c0ec4515e6484ad3e7195dc3f015e8b;p=python.git

Initial commit.
---

0db243241c0ec4515e6484ad3e7195dc3f015e8b
diff --git a/covid19.py b/covid19.py
new file mode 100755
index 0000000..0fd9ecc
--- /dev/null
+++ b/covid19.py
@@ -0,0 +1,70 @@
+#!/usr/bin/env python
+
+import os, time, math
+import numpy, csv
+import matplotlib.pyplot as plt
+import matplotlib.dates as mdates
+import urllib.request
+
+url = 'https://raw.githubusercontent.com/CSSEGISandData/COVID-19/master/csse_covid_19_data/csse_covid_19_time_series/time_series_19-covid-Confirmed.csv'
+
+file = 'time_series_19-covid-Confirmed.csv'
+
+######################################################################
+
+if not os.path.isfile(file) or os.path.getmtime(file) < time.time() - 86400:
+    print('Retrieving file')
+    urllib.request.urlretrieve(url, file)
+
+######################################################################
+
+with open(file, newline='') as csvfile:
+    reader = csv.reader(csvfile, delimiter=',')
+    times = []
+    nb_cases = {}
+    time_col = 5
+    for row_nb, row in enumerate(reader):
+        for col_nb, field in enumerate(row):
+            if row_nb >= 1 and col_nb == 1:
+                country = field
+                if not country in nb_cases:
+                    nb_cases[country] = numpy.zeros(len(times))
+                # print(country)
+            if row_nb == 0 and col_nb >= time_col:
+                times.append(time.mktime(time.strptime(field, '%m/%d/%y')))
+            if row_nb == 1 and col_nb == time_col:
+                nb_cases['World'] = numpy.zeros(len(times))
+            if row_nb >= 1:
+                if col_nb >= time_col:
+                    nb_cases['World'][col_nb - time_col] += int(field)
+                    nb_cases[country][col_nb - time_col] += int(field)
+
+######################################################################
+
+fig = plt.figure()
+ax = fig.add_subplot(1, 1, 1)
+
+ax.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')
+
+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)
+
+# ax.legend(loc='center left', bbox_to_anchor=(1, 0.5), frameon = False)
+ax.legend(frameon = False)
+
+plt.show()
+# fig.savefig('covid19.svg')
+
+######################################################################