本文介绍了 pandas 分组并将行转换为多列的处理方法,对大家解决问题具有一定的参考价值,需要的朋友们下面随着跟版网的小编来一起学习吧!
问题描述
data = {'groupId':[1,1,2], 'email':['a1@gmail.com', 'a2@gmail.com', 'a3@gmail.com'],
'type':['office','personal','personal'],'name':['santy','santy','will']}
df = pd.DataFrame(data)
我有一个这样的数据帧
groupId email type name
1 a1@gmail.com office santy
1 a2@gmail.com personal santy
2 a3@gmail.com personal will
我要根据特定组中的行数将行转换为动态列
groupId email1 type1 email2 type2 name
1 a1@gmail.com office a2@gmail.com personal santy
2 a3@gmail.com 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 a1@gmail.com office a2@gmail.com personal
2 a3@gmail.com personal NaN NaN
这篇关于 pandas 分组并将行转换为多列的文章就介绍到这了,希望我们推荐的答案对大家有所帮助,也希望大家多多支持跟版网!
本站部分内容来源互联网,如果有图片或者内容侵犯了您的权益,请联系我们,我们会在确认后第一时间进行删除!