python - 使用 Seaborn/Matplotlib 的组条形图

Question

我的目标是使用按两个变量“Alpha”和“Beta”分组的 pandas DataFrame 创建如下所示的分组条形图。

xl2 = xl.groupby(['Alpha','Beta']).median()

当我尝试这样做时，在“Alpha”上抛出了一个 KeyError

import seaborn as sns
sns.barplot(x=['Alpha', 'Beta'], y=xl2['Gamma'])

我希望传递一个 x 值列表以作为索引(“Alpha”和“Beta”)，并绘制相关的“Gamma”。seaborn.barplot function 的文档没有提供任何组条形图示例。

感谢您的帮助!

Answer 1

你可以为此使用 ggplot

from ggplot import *
import pandas as pd
import numpy as np

df = pd.DataFrame({
    "x": np.random.choice(range(2001, 2008), 250),
    "w": np.random.uniform(50, 400, 250),
    "cat": np.random.choice(["A", "B", "C", "D", "E"], 250)
})

print ggplot(df, aes(x='x', weight='w', fill='cat')) + geom_bar() + theme_bw()