按特定列分组，列出其他列 Pandas

Question

我正在尝试对数据框中的一列进行分组，并且希望将其他列添加到元素列表中。

我有一个与此类似的数据框：

Route  Station  Position
A1      X1         P1        
A1      X2         P2        
A1      X3         P3      
B2      Y1         P1   
B2      Y2         P2

预期的输出应该是这样的：

Route  Station  Position
A1   [X1,X2,X3] [P1,P2,P3]       
B2    [Y1,Y2]    [P1,P2]   

正确的做法是什么？

Answer 1

使用：

df.groupby('Route', as_index=False).agg(list)

输出：

  Route       Station      Position
0    A1  [X1, X2, X3]  [P1, P2, P3]
1    B2      [Y1, Y2]      [P1, P2]