如何根據Python中的列值聚合某些行

Question

如果在數據框的特定列中共享相同的值，我是否要聚合某些列中的值？

換句話說，我怎麼能從A獲取數據幀B？在這個例子中，我想根據列c1中的值進行檢查，如果存在具有相同值的行，那麼我想將列c3-c5的總和放在行中新列（c6）中的值爲「 c「列c2。

答：

c0 c1 c2 c3 c4 c5 
0 1 a d 3 4 0 
1 1 a c 0 0 6 
2 1 b d 3 1 0 
1 1 b c 0 0 1 

B：

c0 c1 c2 c3 c4 c5 c6 
0 1 a d 3 4 0 
1 1 a c 0 0 6 13 
2 1 b d 3 1 0 
1 1 b c 0 0 1 5 

這裏是我做了什麼：

import pandas as pd 
import numpy as np 
A = pd.read_excel(file_location) 
A['c6'] = pd.Series(0, index=A.index) 
A.c6[A['c2'] == 'c'] = A.c5 + A.c4[A.c1 == A.c1.shift(-1)] + A.c3[A.c1 == A.c1.shift(-1)] 
B = A 

Answer 1

import pandas as pd 
import io 
import numpy as np 
import datetime 

data = """ 
    c0 c1 c2 c3 c4 c5 
0 1 a d 3 4 0 
1 1 a c 0 0 6 
2 1 b d 3 1 0 
1 1 b c 0 0 1 
    """ 

df = pd.read_csv(io.StringIO(data), delimiter='\s+') 

df2 = pd.DataFrame(df.groupby('c1').sum().ix[:,1:].sum(axis=1), columns=['c6']).reset_index() 

df3 = pd.merge(df,df2,on='c1').set_index(df.index) 

print(df3) 

這將使：

c0 c1 c2 c3 c4 c5 c6 
0 1 a d 3 4 0 13 
1 1 a c 0 0 6 13 
2 1 b d 3 1 0 5 
1 1 b c 0 0 1 5