python - 在保持相似的 DataFrame 结构的同时汇总 DataFrame 中的行

Question

我有以下数据框:

Stint    Year   ID  Data1    Data2  Team
  1     2010    A    10       1      SFN
  1     2011    A    10       1      SFN
  1     2013    A    10       1      SFN
  2     2013    A    10       1      ATL
  1     1922    B    10       1      ARI
  1     1923    B    10       1      ARI
  1     1924    B    10       1      ARI

我正在尝试返回一个新的 DataFrame，它将相同年份的 Data1 和 Data2 列中的值相加。我希望上面的 DataFrame 最终看起来像这样:

    Year   ID  Data1    Data2 
    2010    A    10       1      
    2011    A    10       1         
    2013    A    20       2      
    1922    B    10       1      
    1923    B    10       1      
    1924    B    10       1      

我搞砸了一些 groupby 函数，但我无法获得正确的 DataFrame 结构。

谢谢!

Answer 1

groupby 与 as_index=False

不会在新索引中包含分组列

df.groupby(['Year', 'ID'], as_index=False)[['Data1', 'Data2']].sum()

   Year ID  Data1  Data2
0  1922  B     10      1
1  1923  B     10      1
2  1924  B     10      1
3  2010  A     10      1
4  2011  A     10      1
5  2013  A     20      2