pandas 分组并将行转换为多列 [英] pandas group by and convert rows into multiple columns

问题描述

data = {'groupId':[1,1,2], 'email':['[e-mail]', '[e-mail]', '[e-mail]'],
        'type':['office','personal','personal'],'name':['santy','santy','will']} 
df = pd.DataFrame(data) 

我有一个这样的数据帧

groupId email   type           name
1   [e-mail]    office      santy
1   [e-mail]    personal    santy
2   [e-mail]    personal    will

我要根据特定组中的行数将行转换为动态列

groupId email1         type1   email2          type2       name
1      [e-mail]  office    [e-mail]    personal    santy
2      [e-mail]   personal   na              na        will

我知道可以将SET_INDEX与UNSTACK一起使用，但是不知道如何才能给出列名并创建特定组中的那么多列。

有没有有效的方法做到这一点？ 如有任何帮助，我们将不胜感激

推荐答案

您可以执行以下操作：

new_df = (df.assign(col=df.groupby('groupId').cumcount()+1)
   .set_index(['groupId','col'])
   .unstack('col')
   .sort_index(level=(1,0), axis=1)
)

new_df.columns = [f'{x}{y}' for x,y in new_df.columns]

输出：

               email1     type1        email2     type2
groupId                                                
1        [e-mail]    office  [e-mail]  personal
2        [e-mail]  personal           NaN       NaN

这篇关于 pandas 分组并将行转换为多列的文章就介绍到这了，希望我们推荐的答案对大家有所帮助，也希望大家多多支持IT屋！