python - 创建按一列和新列分组的新数据框

Question

import pandas as pd
import numpy as np

df = pd.DataFrame( {
   'A': ['d','d','d','f','f','f','g','g','g','h','h','h'],
   'B': [5,5,6,7,5,6,6,7,7,6,7,7],
   'C': [1,1,1,1,1,1,1,1,1,1,1,1],
   'S': [2012,2013,2014,2015,2016,2012,2013,2014,2015,2016,2012,2013]     
    } );

df

我想创建一个新的数据框，其中 A 列进行分组，B 列和 C 列的总和显示在新列 2012、2013、2014、2015 和 2016 中。

Answer 1

您可以使用groupby与 unstack :

df = (df.B + df.C).groupby([df.A, df.S]).sum().unstack(fill_value=0)
print (df)
S  2012  2013  2014  2015  2016
A                              
d     6     6     7     0     0
f     7     0     0     8     6
g     0     7     8     8     0
h     8     8     0     0     7

<小时/>

df['D'] = df.B + df.C

df = df.groupby(['A','S'])['D'].sum().unstack(fill_value=0)
print (df)
S  2012  2013  2014  2015  2016
A                              
d     6     6     7     0     0
f     7     0     0     8     6
g     0     7     8     8     0
h     8     8     0     0     7

另一个解决方案 pivot_table :

df['D'] = df.B + df.C
df = df.pivot_table(index='A', columns='S', values = 'D', aggfunc='sum', fill_value=0)
print (df)
S  2012  2013  2014  2015  2016
A                              
d     6     6     7     0     0
f     7     0     0     8     6
g     0     7     8     8     0
h     8     8     0     0     7