使用sm.tsa.ARIMA的方法生成样本数据。下面是代码示例:
import numpy as np
import statsmodels.api as sm
# 定义ARIMA模型参数
order = (1, 1, 1)
ar_coefs = [0.8]
ma_coefs = [0.2]
# 创建ARIMA模型
arima_mod = sm.tsa.ARIMA(endog=np.zeros(20), order=order)
arima_fit = arima_mod.fit()
# 生成样本数据
sample_data = arima_fit.predict(start=0, end=19, start_ar_lags=1, end_ar_lags=1, start_ma_lags=1, end_ma_lags=1, dynamic=False)
print(sample_data)