pandas 分组并将行转换为多列

本文介绍了 pandas 分组并将行转换为多列的处理方法，对大家解决问题具有一定的参考价值，需要的朋友们下面随着小编来一起学习吧！

问题描述

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

我有一个这样的数据帧

groupId email   type           name
1   [email protected]    office      santy
1   [email protected]    personal    santy
2   [email protected]    personal    will

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

groupId email1         type1   email2          type2       name
1      [email protected]  office    [email protected]    personal    santy
2      [email protected]   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        [email protected]    office  [email protected]  personal
2        [email protected]  personal           NaN       NaN

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