熊猫合并成名单群
熊猫合并成名单群
提问于 2017-09-20 06:25:59
我有这样的数据
id sid frame type
1 10 15 person
2 11 45 person
2 11 45 animal
3 13 75 person
4 14 45 person 我的目标是在这个文件中转换数据。
id sid frame type
1 10 15 person
2 11 45 [person, animal]
3 13 75 person
4 14 45 person我正在寻找一种将type列与同一id, sid, frame合并的方法
回答 1
Stack Overflow用户
回答已采纳
发布于 2017-09-20 06:28:18
您可以将groupby与条件一起使用list：
df = (df.groupby(['id','sid','frame'])['type']
.apply(lambda x: x.tolist() if len(x) > 1 else x.values[0])
.reset_index())
print (df)
id sid frame type
0 1 10 15 person
1 2 11 45 [person, animal]
2 3 13 75 person
3 4 14 45 person如果所有值都应该是lists：
df = (df.groupby(['id','sid','frame'])['type']
.apply(list)
.reset_index())
print (df)
id sid frame type
0 1 10 15 [person]
1 2 11 45 [person, animal]
2 3 13 75 [person]
3 4 14 45 [person]页面原文内容由Stack Overflow提供。腾讯云小微IT领域专用引擎提供翻译支持
原文链接：
https://stackoverflow.com/questions/46314708
复制相关文章
相似问题
腾讯云开发者
