python - 如何对pandas数据框中的复杂条件求和

Question

我有下面的数据框；

df=pd.DataFrame(np.arange(1,19).reshape(6,3),columns=list('ABC'),index=list('acbabc'))

    A   B   C
a   1   2   3
c   4   5   6
b   7   8   9
a  10  11  12
b  13  14  15
c  16  17  18

我想要条件求和数据框，如下所示；

   A  B  C
a 11 13 15
b 20 22 24
c 20 22 24

每个元素显示 df 的条件和。例如，(我对表达式没有信心)

result.loc[0,0]=df.loc[df.A=="a"].sum() 

如何获取此数据框？

Answer 1

Groupby index 和 sum 这些列应该可以满足您的需求:

df.groupby(df.index).sum()

#    A   B   C
#a  11  13  15
#b  20  22  24
#c  20  22  24