重塑pandas中的数据帧

时间:2014-02-06 22:10:38

标签: python pandas dataframe reshape

pandas中有一种简单的方法可以重塑以下数据框:

df = pd.DataFrame({'n':[1,1,2,2,1,1,2,2],
                   'l':['a','b','a','b','a','b','a','b'],
                   'v':[12,43,55,19,23,52,61,39],
                   'g':[0,0,0,0,1,1,1,1]
                  })

这种格式?:

   g  a1 b1 a2 b2
   0  12 43 55 19
   1  23 52 61 39

1 个答案:

答案 0 :(得分:3)

In [75]: df['ln'] = df['l'] + df['n'].astype(str)

In [76]: df.set_index(['g', 'ln'])['v'].unstack('ln')
Out[76]: 
ln  a1  a2  b1  b2
g                 
0   12  55  43  19
1   23  61  52  39

[2 rows x 4 columns]

如果您需要订购,那么:

In [77]: df.set_index(['g', 'ln'])['v'].unstack('ln')[['a1', 'b1', 'a2', 'b2']]
Out[77]: 
ln  a1  b1  a2  b2
g                 
0   12  43  55  19
1   23  52  61  39

[2 rows x 4 columns]