EmpezarEmpieza gratis

Generación de datos ARMA

En este ejercicio vas a generar datos AR/MA/ARMA para 100 días. Recuerda que, en aplicaciones reales, estos datos podrían ser los cambios en el precio de las acciones de Google, las necesidades energéticas de la ciudad de Nueva York o el número de casos de gripe.

Puedes usar la función arma_generate_sample() disponible en tu espacio de trabajo para generar series temporales usando distintos coeficientes AR y MA.

Recuerda que, para cualquier modelo ARMA(p,q):

  • La lista ar_coefs tiene la forma [1, -a_1, -a_2, ..., -a_p].
  • La lista ma_coefs tiene la forma [1, m_1, m_2, ..., m_q],

donde a_i son los coeficientes AR del retardo i y m_j son los coeficientes MA del retardo j.

Este ejercicio forma parte del curso

Modelos ARIMA en Python

Ver curso

ejercicio interactivo práctico

Prueba este ejercicio completando este código de ejemplo.

# Import data generation function and set random seed
from statsmodels.tsa.arima_process import arma_generate_sample
np.random.seed(1)

# Set coefficients
ar_coefs = [____]
ma_coefs = [____]

# Generate data
y = arma_generate_sample(____, ____, nsample=____, scale=0.5)

plt.plot(y)
plt.ylabel(r'$y_t$')
plt.xlabel(r'$t$')
plt.show()
Editar y ejecutar código