将与唯一标签对应的值附加为熊猫数据框中的值列表

Question

我有一个数据框：

df  =      A    B
       0   a    1
       1   b    2
       2   a    3 
       3   d    4

我想像这样更新它：

df  =      A    B
       0   a    [1, 3]
       1   b    [2] 
       2   d    [4]

Answer 1

您可以groupby列A并将B中的分组元素转换为带有apply的列表：

df.groupby('A').B.apply(list).reset_index()

   A       B
0  a  [1, 3]
1  b     [2]
2  d     [4]