大熊猫据帧重塑/枢

Question

我想转换下面大熊猫据帧

a b 
0 1 2 
1 1 5 
2 2 4 
3 1 3 
4 3 7 
5 2 1 

到

0 1 2 
a   
1 2 5 3 
2 4 1 NaN 
3 7 NaN NaN 

你知道一个简单的方法？

Answer 1

我会做如下：

In [11]: df.groupby("a")["b"].apply(lambda x: pd.Series(x.values)) 
Out[11]: 
a 
1 0 2 
    1 5 
    2 3 
2 0 4 
    1 1 
3 0 7 
Name: b, dtype: int64 

得到你想要的，那么你拆散形式（虽然可能以上更好）：

In [22]: df.groupby('a')["b"].apply(lambda x: pd.Series(x.values)).unstack(1) 
Out[22]: 
    0 1 2 
a 
1 2.0 5.0 3.0 
2 4.0 1.0 NaN 
3 7.0 NaN NaN