covid19-linreg.py

import numpy as np
import pandas as pd
from sklearn import linear_model
import matplotlib.pyplot as plt

# Carregando os dados
data_owd = pd.read_csv('https://covid.ourworldindata.org/data/owid-covid-data.csv')

# Filtrando os dados (Canadá)
can = data_owd[data_owd.iso_code == 'CAN']
can = can[can.new_cases > 50]
can = can[~np.isnan(can.new_tests)]

# Ajustando a curva
x = can.new_tests.values.reshape(-1, 1)
y = can.new_cases.values.reshape(-1, 1)

regr = linear_model.LinearRegression()
regr.fit(x, y)
y_pred = regr.predict(x)

plt.figure(dpi = 150)
plt.scatter(x, y)
plt.plot(x, y_pred, color='red')
plt.show()