如何在数据框中加入索引的公共元素?

时间:2017-02-25 18:05:09

标签: python pandas dataframe

我有一个这样的数据框:

           type count
batsman         
V Kohli     4   361
V Kohli     6   149
SK Raina    5   1
SK Raina    6   161
RG Sharma   5   1
RG Sharma   6   164

现在我们可以看到索引中有commmon条目。我想绘制一个 barh plot ,所以我想将索引编入一个,以便数据框看起来像这样:

            type count
batsman         
V Kohli     4   361
            6   149
SK Raina    5   1
            6   161
RG Sharma   5   1
            6   164

我该怎么做?

2 个答案:

答案 0 :(得分:0)

您可以将索引分配给新列并执行分组。

df['batsman'] = df.index
grouped_df = df.groupby(['batsman', 'type']).agg({'count': 'sum'})
# grouped_df

           type count
batsman         
V Kohli     4   361
            6   149
SK Raina    5   1
            6   161
RG Sharma   5   1
            6   164

答案 1 :(得分:0)

您可以groupby您的DataFrame索引和"输入"然后聚合以在您的barh图中获得合理的分组输出。

df.groupby([df.index,'type'])['count'].sum().plot.barh(title='Counts')

enter image description here