AR(p) model in python