Analyse des séries chronologiques en Python
Rob Reider
Adjunct Professor, NYU-Courant Consultant, Quantopian
Estimer des paramètres à partir de données (simulées)
from statsmodels.tsa.arima_model import ARMA
mod = ARMA(data, order=(1,0))
result = mod.fit()
ARMA est obsolète et remplacé par ARIMA
from statsmodels.tsa.arima.model import ARIMA
mod = ARIMA(data, order=(1,0,0))
result = mod.fit()
Pour ARMA, order=(p,q)
print(result.summary())

print(result.params)
array([-0.03605989, 0.90535667])
from statsmodels.graphics.tsaplots import plot_predict
fig, ax = plt.subplots()
data.plot(ax=ax)
plot_predict(result, start='2012-09-27', end='2012-10-06', alpha=0.05, ax=ax)
plt.show()
plot_predict()alpha=None pour aucun intervalle de confianceax=ax pour tracer les données et la prévision sur les mêmes axes
Analyse des séries chronologiques en Python