如何将数据框单元格中的分隔行组合成列表[重复]

Question

我希望通过将用户 ID 匹配到列表中来组合数据框中的熊猫行。每个有 1000 多个用户 ID 和多个条目。

我想为每个用户一行。 我发现了一个与我想要做的完全相反的线程，即：How to explode a list inside a Dataframe cell into separate rows 提前谢谢各位。

对于带有字符串和数字的行需要做什么？

  userID    alcohol 
  U1001     No_Alcohol_Served 7.0
            Wine-Beer 2.0
  U1002     Full_Bar 1.0
            No_Alcohol_Served 3.0
            Wine-Beer 6.0
  U1003     Full_Bar 2.0
            No_Alcohol_Served 8.0
            Wine-Beer 3.0
  U1004    No_Alcohol_Served 4.0
           Wine-Beer 4.0

我想说的是：

U1001 : No_Alcohol_served:7.0, Wine-Beer:2.0
U1002 : Full_Bar:1.0, No_Alcohol_served:3.0, Wine_beer:6.0

以此类推

Answer 1

你可以试试这样的：

df.groupby('userID').apply(lambda x: x['name'].tolist())

示例：

给定一个df：

  userID name
0  U1001    a
1  U1001    b
2  U1001    c
3  U1002    d
4  U1002    e
5  U1003    f

>>> df.groupby('userID').apply(lambda x: x['name'].tolist())
userID
U1001    [a, b, c]
U1002       [d, e]
U1003          [f]
dtype: object