我正在尝试转换以下数据框:
cat userid val
1 1 10
1 2 15
1 3 22
2 1 33
2 2 54
2 3 44
采用以下格式:[[10, 15, 22], [33, 54, 44]]
这是我的代码:
df.groupby(['cat', 'userid'])['val'].apply(list)
而且,这就是我得到的:
cat userid
1 1 [10]
2 [15]
3 [22]
2 1 [33]
2 [54]
3 [44]
有什么建议吗?