Pandas在多次聚合后保留列

时间:2015-12-06 13:57:11

标签: python pandas

我试图对熊猫数据框进行多次恶化,问题是我想保留列聚合

df3 = pd.DataFrame({'X' : ['A', 'B', 'A', 'B'], 'Y' : [1, 4, 3, 2]})
df3.groupby('X', as_index=False).agg('sum')

   X  Y
0  A  4
1  B  6

这很好,但我想要的是像这样的多个聚合

df3 = pd.DataFrame({'X' : ['A', 'B', 'A', 'B'], 'Y' : [1, 4, 3, 2]})
df3.groupby('X', as_index=False).agg(['sum', 'mean'])

它给了我

    Y     
    sum mean
X         
A   4    2
B   6    3

但我想要这个

    X   Y     
        sum mean

0   A   4   2
1   B   6   3

1 个答案:

答案 0 :(得分:1)

要将reset_index从索引移至列,请使用In [4]: df3 = pd.DataFrame({'X' : ['A', 'B', 'A', 'B'], 'Y' : [1, 4, 3, 2]}) In [5]: df3.groupby('X', as_index=False).agg(['sum', 'mean']).reset_index() Out[5]: X Y sum mean 0 A 4 2 1 B 6 3

{{1}}