pycoro

#!/usr/bin/env python3
import urllib.request
import json
from datetime import datetime
from scipy.interpolate import UnivariateSpline
from scipy.signal import savgol_filter


if __name__ == "__main__":
    from matplotlib import pyplot as plt

    firstDay = datetime(2020, 2, 24)

    url = "https://raw.githubusercontent.com/pcm-dpc/COVID-19/master/dati-json/dpc-covid19-ita-regioni.json"
    req = urllib.request.Request(url)
    with urllib.request.urlopen(req) as response:
        datastore = json.loads(response.read())
    for regione in ['Veneto','Friuli Venezia Giulia']:
        xPoints = []
        yPoints = []
        xMax=-1
        for data in datastore:
            if data['denominazione_regione']==regione:
                date_time_obj = datetime.strptime(data['data'], '%Y-%m-%d %H:%M:%S')
                delta=date_time_obj-firstDay
                xMax = max(xMax, delta.days)
                xPoints.append(delta.days)
                yPoints.append(data['totale_casi'])
        plt.plot(xPoints, yPoints, 'o',label='data')
        yhat = savgol_filter(yPoints, 11, 3)
        plt.plot(xPoints, yhat,label='Savitzky–Golay')
        daysWanted=[xMax]
        for i in range(0, 4):
            daysWanted.append(xMax+i)

        for k in (1, 2,3):
            spl = UnivariateSpline(xPoints, yhat, k=k)
            spl.set_smoothing_factor(0.2)
            y = spl(daysWanted)
            label = "k=%d" % k
            print(label,y)
            plt.plot(daysWanted, y, '--', label=label)
        plt.grid()
        plt.title(regione)
        plt.legend( loc="best" )
        plt.show()