python - Pandas 合并具有相同 UserID 的行

Question

我有代表帖子的数据，其中包含有关发布该帖子的用户以及与其关联的标签的信息。

          UserID      PostID  Tags   
0           1         1        A   
1           2         2        A    
2           1         3        A    
3           2         4        B   
4           3         5        B    
5           4         6        B    
6           5         7        A   

我想创建一个新的数据框，其中每一行代表一个像这样的用户。

          UserID      Posts     Tags   
0           1         [1,3]     [A,A]   
1           2         [2,4]     [A,B]   
2           3         [5]       [B]    
3           4         [6]       [B]   
4           5         [7]       [A]    

这样我就可以获得帖子和标签的列表，而不是让它们各自独立。

有什么想法吗？

Answer 1

将 groupby.agg 与 list 一起使用:

df.groupby('UserID', as_index=False)['PostID', 'Tags'].agg(lambda x: list(x))
Out: 
   UserID  PostID    Tags
0       1  [1, 3]  [A, A]
1       2  [2, 4]  [A, B]
2       3     [5]     [B]
3       4     [6]     [B]
4       5     [7]     [A]