Adicionar barras de erro a valores previstos em python

Question

Executei um modelo lstm 10 vezes e calculei a média dos valores para obter a média, que se tornou meu valor previsto. Calculei a divisão padrão de cada linha e a adicionei ao meu dataframe (test_iterations) que contém meus valores previstos. Agora quero traçar barras de erro para que meu gráfico fique assim (isso foi feito no Excel)

Mas quando eu uso o código a seguir, obtenho um gráfico parecido com este

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

Não tenho certeza se o que estou fazendo é certo porque adicionei os valores xey no código da mesma coisa, mas não tinha certeza do que adicionar lá

Answer 1

O primeiro argumento de plt.errorbar()são as coordenadas x dos pontos a serem plotados. No seu caso, é a classificação do valor.

Experimente isto:

import numpy as np

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