python - 根据dataframe split得到一系列数据

Question

我有一个像这样的 pandas 数据框:

Cat | Val
 A  | 10
 B  | 20
 B  | 5
 A  | 8

我想将列拆分为

[[10, 8],[20,5]]
[A,B]

我知道这非常简单，只是找不到合适的在线资源

Answer 1

通过使用groupby apply和list

s=df.groupby('Cat').Val.apply(list)
s.index.tolist()
Out[273]: ['A', 'B']
s.values.tolist()
Out[274]: [[10, 8], [20, 5]]

s.reset_index()
Out[276]: 
  Cat      Val
0   A  [10, 8]
1   B  [20, 5]