基于Python中列名的数据框到列表列表的转换

Question

输入：
数据框

col1    col2
item1   10
item1   20
item1   25
item2   56
item2   36
item3   1

输出：
列表清单

[[10,20,25],[56,36],[1]]

Answer 1

使用GroupBy.apply由Series创建lists中的groupby，最后将其转换为list：

L = df.groupby('col1')['col2'].apply(list).tolist()
print (L)
[[10, 20, 25], [56, 36], [1]]

Answer 2

您可以GroupBy col1并从col2的每一组中提取列表：

df.groupby('col1').col2.apply(list).values.tolist()
# [[10, 20, 25], [56, 36], [1]]