使用列的唯一值作为索引对数据框进行分组

Question

What is the most efficient way of group or stack the different values of dataframe indexing by value?

this df:

color   alert

blue    A
blue    B
red     A
green   C

to:

color   alert

blue    [A, B]
red     A
green   C

Answer 1

You need a groupby and to aggregate into a list :

df.groupby('color').alert.agg(list).reset_index()

   color   alert
0   blue  [A, B]
1  green     [C]
2    red     [A]