Skip to main content
 首页 » 编程设计

python之Pandas groupby 年份和绘图图

2025年05月04日23linjiqin

这与“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: 

请您参考如下方法:

让我们试试这段代码:

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