Преобразование информационного кадра pandas в список пар ключ-значение

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']}

Пожалуйста, предложите, как это сделать?

Answer 1

Раствор от itertools

import itertools

l=df.sort_values('col1').values.tolist()
{k: [x[1] for x in g] for k, g in itertools.groupby(l,lambda x : x[0])}
{'a': [11, 14, 15], 'b': [12, 13]}

Answer 2

Группировка по первому столбцу, сбор значений второго столбца в списки, преобразование результата в словарь:

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