python - 在 python 中向预测值添加误差线

Question

我运行了 lstm 模型 10 次，并对这些值进行平均以获得平均值，这就是我的预测值。我计算了每行的标准偏差并将其添加到包含我的预测值的数据帧(test_iterations)中。我现在想绘制误差线，以便我的图表看起来像这样(在 Excel 中完成)

但是当我使用以下代码时，我得到一个如下所示的图表

plt.errorbar(test_iterations['Mean'].values, 
                     test_iterations['Mean'].values,
                     yerr = test_iterations['Stdv'].values,
                     fmt='-o')

我不确定我所做的是否正确，因为我在同一件事的代码上添加了 x 和 y 值，但不确定要添加什么

Answer 1

plt.errorbar() 的第一个参数是要绘制的点的 x 坐标。在您的情况下，它是值的排名。

试试这个:

import numpy as np

plt.errorbar(np.arange(len(test_iterations['Mean'])),
             test_iterations['Mean'].values,
             yerr=test_iterations['Stdv'].values,
             fmt='-o')