熊猫组和总和值

时间:2017-04-11 07:44:55

标签: python python-3.x csv pandas

这是我从CSV导入的示例表

Name    hours   work
User1   2       Kiwi
User1   5       Melon
...
User1   3       Kiwi

这是我想要的输出:

Total Kiwi:  
User1   5   

我猜可以使用正确的联接或分组。但我不能把它重新放到真正的代码上。 我试过这样的事情

ou = pd.DataFrame([[ou["work"].sum()["kiwi"]]])

1 个答案:

答案 0 :(得分:2)

你需要:

df = df.groupby(['Name','work'])['hours'].sum().unstack()
print (df)
work   Kiwi  Melon
Name              
User1     5      5

或者:

df = df.pivot_table(index='Name', columns='work', values='hours', aggfunc='sum')
print (df)

work   Kiwi  Melon
Name              
User1     5      5

然后:

print (df[['Kiwi']])
work   Kiwi
Name       
User1     5