python - 将熊猫数据框转换为键值对列表

Question

我有一个熊猫数据框，如下所示：

 col1 col2 
0  a     11
1  b     12
2  b     13
3  a     14
4  a     15 

我想把它转换成一个应该像的列表

myList = {'a':['11','14','15'],'b':['12','13']}

请建议怎么做？

标签： pythonpandasdataframe

Answer 1

按第一列分组，将第二列的值收集到列表中，将结果转换为字典：

df.groupby('col1')['col2'].apply(list).to_dict()
#{'a': ['11', '14', '15'], 'b': ['12', '13']}