如何在pandas数据帧中总结复杂的条件

时间:2016-08-31 02:08:51

标签: python pandas numpy dataframe

我有以下数据框;

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() 

如何获取此数据框?

1 个答案:

答案 0 :(得分:4)

Groupby indexsum列可以为您提供所需内容:

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

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