python - 带 2 列的条形图

Question

我想使用以下数据框创建水平条形图

studentData = { 
0 : {
    'stylevel' : 'bachelor',
    'q' : 'agree',
    'Type' : 'Nascent'
    },
1 : {
    'stylevel' : 'bachelor',
    'q' : 'very',
    'Type' : 'Nascent'
    },
2 : {
    'stylevel' : 'master',
    'q' : 'very',
    'Type' : 'Active'
    },    
3 : {
    'stylevel' : 'master',
    'q' : 'agree',
    'Type' : 'Active'
    },  
4 : {
    'stylevel' : 'master',
    'q' : 'agree',
    'Type' : 'Student'
    },
}

dfObj = pd.DataFrame(studentData)
dfObj = dfObj.T

我正在使用以下代码绘制图形

dfObj['q'].value_counts().plot(kind='barh')

您能告诉我如何使用原始数据框中的 Type 列创建相同的图形以将 q 数据分成 3 个子组(即学生、事件和新生)吗？

Answer 1

你有三个选择。
使用 Pandas :

dfObj.groupby('Type')['q'].value_counts().plot(kind='barh')

使用 Pandas 堆叠条:

dfObj.groupby('Type')['q'].value_counts().unstack(level=0).plot.barh(stacked=True)

使用 seaborn.catplot :

import seaborn as sns
df2 = dfObj.groupby('Type')['q'].value_counts().rename('count').reset_index()
sns.catplot(data=df2, x='q', hue='Type', y='count', kind='bar')