按类别对数据框进行分组,并将结果转换为嵌套列表

时间:2017-09-30 13:22:30

标签: python pandas dataframe group-by

我正在尝试转换以下数据框:

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]

有什么建议吗?

0 个答案:

没有答案