重塑熊猫据帧分组变量

Question

我有以下格式

ID Name 
0 1 Jim 
1 1 Jimmy 
2 2 Mark 
3 2 Marko 
4 3 Sergi 
4 3 Sergi 

我要重塑数据帧的格式如下

ID Name_1 Name_2 
0 1  Jim Jimmy 
1 2 Mark Marko 
2 3 Sergi Sergi 

这样我可以比较两个名称的熊猫数据帧。对于此要求，我无法使用pd.pivot或pd.pivottable。 应该相当简单。请，你能建议如何做到这一点？

Answer 1

您可以使用cumcount与pivot，最后add_prefix列名：

df['groups'] = df.groupby('ID').cumcount() + 1 
df = df.pivot(index='ID', columns='groups', values='Name').add_prefix('Name_') 
print (df) 
groups Name_1 Name_2 
ID     
1   Jim Jimmy 
2  Mark Marko 
3  Sergi Sergi 

与groupby和unstack另一种解决方案，最后add_prefix列名：

df1 = df.groupby('ID')["Name"] \ 
     .apply(lambda x: pd.Series(x.values)) \ 
     .unstack(1) \ 
     .rename(columns=lambda x: x+1) \ 
     .add_prefix('Name_') 
print (df1) 
    Name_1 Name_2 
ID    
1  Jim Jimmy 
2 Mark Marko 
3 Sergi Sergi