pandas dataframe groupby sum index

时间:2016-11-04 14:56:35

标签: pandas dataframe

我有一个数据框,我想

    FROM:  
             dow    yield
           0    F   2
           1    F   3
           2    M   4
           3    M   6
           4    TH  7
    TO:

          dow    ysum
      0     F   5
      1     M   10
      2     TH  7

但我得到了这个:

               |yield
         -------------
          dow  |
         -------------
           F   |5
           M   |10
           TH  |7

我就这样做了:

    d1=['F','F','M','M','TH']
    d2=[2,3,4,6,7]
    d = {'dow': d1, 'yield': d2}
    df = pd.DataFrame(data=d, index=None)
    df1= df.groupby('dow').sum()

如何将结果用作列而不是索引?

1 个答案:

答案 0 :(得分:2)

第一列是index,因此您可以添加参数as_index=False

df1 = df.groupby('dow', as_index=False).sum()
print (df1)
  dow  yield
0   F      5
1   M     10
2  TH      7

reset_index

df1 = df.groupby('dow').sum().reset_index()
print (df1)
  dow  yield
0   F      5
1   M     10
2  TH      7