python - 如何绘制我的数据框(python 和 pandas)

Question

我目前有一个如下所示的数据框 (df):我需要绘制它，使其看起来像这样:

![想要的图][2]

在图表中，列是“索引”，蓝色条是“总计”，橙色条是“Total.1”

每次我尝试这样做时，我都会遇到 KeyErrors，有人能给我指出正确的方向吗，因为我觉得我已经尝试了所有方法，但总是出错

Answer 1

要生成一个简单的水平堆叠条形图很容易，下面的代码将完成任务。

df.set_index('index').plot(stacked=True, kind='barh')
plt.tight_layout()
plt.show()

但是，由于您的标签太长，布局会切断信息，您可能需要对标签进行预处理以使其看起来更好。

---

以下是我推荐的绘图格式:

options = {
    'stacked': True,
    'kind': 'barh',
    'title': 'Medals at the Winter and Summer Olympics'
}

df.rename(columns={'Total': 'Winter Games', 'Total.1': 'Summer Games'}).assign(
    code=df['index'].str.extract(r'\((.*?)\)')
).set_index('code')[['Winter Games', 'Summer Games']].plot(**options)

plt.show()

结果是: