输入:
数据框
col1 col2
item1 10
item1 20
item1 25
item2 56
item2 36
item3 1
输出:
列表列表
[[10,20,25],[56,36],[1]]
最佳答案
通过groupby使用GroupBy.apply创建列表的系列最后将其转换为列表:
L = df.groupby('col1')['col2'].apply(list).tolist()
print (L)
[[10, 20, 25], [56, 36], [1]]
关于python - Python中基于列名的Dataframe到List of List的转换，我们在Stack Overflow上找到一个类似的问题： https://stackoverflow.com/questions/56166179/