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

【问题标题】：Dataframe to List of List conversion based on the name of column in Python基于Python中列名的数据框到列表转换
【发布时间】：2019-05-16 10:15:07
【问题描述】：

输入：
数据框

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

输出：
列表列表

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

【问题讨论】：

【解决方案1】：

用groupby 和GroupBy.apply 创建lists 的Series 并最后将其转换为list：

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

【讨论】：

【解决方案2】：

您可以GroupBycol1 并从col2 的每组中提取列表：

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

【讨论】：