python - 通过在 pandas 中分组和添加值来展平列

Question

我有一个像这样的数据框

 id, index, name, count1, count2
   1,   1,  foo, 12, 10
   1, 2,  foo,  11, 12
   1, 3, foo, 23, 12
   1, 1, bar, 11, 21
   ...
   2, 1, foo, ...

我想获取如下数据框

id, name, count1, count2
1, foo, 46,34
1, bar, ..

所以基本上，我想从这个字段中“冲走”索引..同时添加 count1 和 count2 列

如何在 pandas/python 中执行此操作？

Answer 1

这就是你想要的吗？

In [24]: df.groupby(['id','name']).sum().reset_index()
Out[24]:
   id name  index  count1  count2
0   1  bar      1      11      21
1   1  foo      6      46      34

如果您想删除index列:

In [26]: df.groupby(['id','name']).sum().reset_index().drop('index', 1)
Out[26]:
   id name  count1  count2
0   1  bar      11      21
1   1  foo      46      34

数据:

In [25]: df
Out[25]:
   id  index name  count1  count2
0   1      1  foo      12      10
1   1      2  foo      11      12
2   1      3  foo      23      12
3   1      1  bar      11      21