Question

我正在绘制ARIMA模型，但是我有3个SKU，但是在执行循环后我只能得到一个图。请帮助

for sku in sku_list:
    print()
    print(sku)
    df_sku = df[df['Product SKU'].isin([sku])]
    period_index = 0
    y = df_sku.Sales
    model = ARIMA(y, order=(2, 1, 0))
    results_ARIMA = model.fit(disp=-1)  
    plt.plot(y)
    plt.plot(results_ARIMA.fittedvalues, color='red')
    plt.title('RSS: %.4f'% sum((results_ARIMA.fittedvalues-y)**2))
    plt.show()

Answer 1

您是否尝试过更改缩进并创建不同的数据集？您的循环似乎只返回一个值。

y=[]
results_ARIMA=[]
for sku in sku_list:
    print()
    print(sku)
    df_sku = df[df['Product SKU'].isin([sku])]
    period_index = 0
    y.append(df_sku.Sales)
    model = ARIMA(y, order=(2, 1, 0))
    results_ARIMA.append(model.fit(disp=-1).fittedvalues)
plt.plot(y)
plt.plot(results_ARIMA, color='red')
plt.title('RSS: %.4f'% sum((results_ARIMA-y)**2))
plt.show()

Python matplotlib并未针对所有3个可用选项进行绘图

1 个答案: