python - 将对应于唯一标签的值附加为 pandas 数据框中的值列表

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]