python-3.x - 带有来自Dataframe的标准错误的条形图？

Question

我有一个存储回归结果的 DataFrame，如下所示：

feats = ['X1', 'X2', 'X3']
betas = [0.5, 0.7, 0.9]
ses = [0.05, 0.03, 0.02]
data = {
    "Feature": feats, 
    "Beta": betas,
    "Error":ses
}

data = pd.DataFrame(data)

它看起来像这样：

   Beta  Error Feature
0   0.5   0.05      X1
1   0.7   0.03      X2
2   0.9   0.02      X3

我想为每个特征制作一个图形系数，高度为“Beta”，误差线为“Error”。

有没有办法让它在 Matplot 中工作？

我已经尝试过错误情节，但可能做错了什么。

标签： python-3.xpandasmatplotlibseaborn

Answer 1

您可以使用plt.errorbar如下 ( matplotlib 2.2.2)

plt.errorbar(data.Feature, data.Beta, yerr=data.Error, capthick=2, capsize=2) 

如果以上行不适合您，您可以使用此解决方法

plt.errorbar(range(len(data.Feature)), data.Beta, yerr=data.Error, capthick=2, capsize=2)
plt.xticks(range(len(data.Feature)), data.Feature)