在对多列上的数据进行分组后绘制堆积条形图

Question

我有一个 Pandas 数据框，它有 2 列，例如 -

Vendor     Product    Category
VendorA    ProdABC       B
VendorA    ProdXYZ       C
VendorAB   ProdCDC       A
VendorAB   ProdDEF       A
VendorAB   ProdKLM       B
VendorF    ProdXYZ       D
VendorC    ProdBSE       C
VendorF    ProdFGH       D
VendorAB   ProdMNO       D
VendorA    ProdFGH       D
VendorV    ProdCDC       A
VendorF    ProdBSE       C

我需要 -

1. 找出列出产品数量最多的前 10 家供应商，并
2. 创建一个堆叠条形图 (matplotlib)，描述前 10 名供应商的产品数量按类别细分，以显示前 10 名供应商销售的产品总数和产品类别的比较。

我该怎么做？

Answer 1

我将使用您在问题中显示的数据框

第一个任务使用groupby.size

top10=df.groupby('Vendor').size().sort_values(ascending=False).head(10)
print(top10)



Vendor
VendorAB    3
VendorA     2
VendorF     1
dtype: int64

---

在top10的vendors（使用：DataFrame.isin）和categories创建组后，使用DataFrame.unstack与DataFrame.plot进行绘图：

top10_by_categories=df[df['Vendor'].isin(top10vendors)].groupby(['Vendor','Category']).count()['Product'].unstack()
categories=top10_by_categories.columns
top10_by_categories['total']=top10_by_categories.sum(axis=1)
top10_by_categories.sort_values(by='total',ascending=False,inplace=True)
print(top10_by_categories)
top10_by_categories[categories].plot(kind='bar',stacked=True)





Category    A    B    C    D  total
Vendor                             
VendorAB  2.0  1.0  NaN  NaN    3.0
VendorA   NaN  1.0  1.0  NaN    2.0
VendorF   NaN  NaN  NaN  1.0    1.0