Statistical rethinking with Python

rethinking_0.4.py

import matplotlib.pyplot as plt
import statsmodels.api as sm
import statsmodels.formula.api as smf

cars = sm.datasets.get_rdataset('cars')
cars = cars.raw_data

model = smf.ols('dist ~ speed', data=cars)
results = model.fit()

print(results.params)

plt.scatter(cars.speed, results.resid)
plt.xlabel('Speed')
plt.ylabel('Residuals')
plt.show()