python - Pandas groupby 年份和绘图图

Question

这与“H-1B Visa Petitions 2011-2016”的 Kaggle 数据集有关:

 [In]: df.groupby(['EMPLOYER_NAME','YEAR']).size().sort_values(ascending=False).head(10)

[Out]:

    EMPLOYER_NAME                      YEAR
    INFOSYS LIMITED                    2015    33245
                                       2013    32223
                                       2016    25352
                                       2014    23759
    CAPGEMINI AMERICA INC              2016    16725
    TATA CONSULTANCY SERVICES LIMITED  2015    16553
    INFOSYS LIMITED                    2012    15818
    TATA CONSULTANCY SERVICES LIMITED  2014    14098
                                       2016    13134
    WIPRO LIMITED                      2015    12201

我想按年份将索引“Infosys Limited”分组在一起，如下所示，因为目前在上面的输出中它单独显示 Infosys Limited 2012 年，还需要绘制图表以了解这些公司在过去 5 年的趋势多年来，我使用 matplotlib 和 seaborn 绘制图表，不确定我们是否可以使用这些工具绘制此类图表。非常感谢任何帮助。

Infosys Limited 2012    15818
                2013    32223
                2014    23759
                2015    33245
                2016    25352

Graph:

Answer 1

让我们试试这段代码:

 df1 = df.groupby(['EMPLOYER_NAME','YEAR']).size().unstack()
 df1.columns = df.columns.droplevel()
 df1.plot(kind='barh')