Панды: сгруппировать по первому столбцу и получить второй столбец в виде списков в строках

Панды: DataFrame Exercise-55 с решением

Напишите программу Pandas для группировки по первому столбцу и получения второго столбца в виде списков в строках.

Пример решения :

Код Python:

import pandas as pd
df = pd.DataFrame( {'col1':['C1','C1','C2','C2','C2','C3','C2'], 'col2':[1,2,3,3,4,6,5]})
print("Original DataFrame")
print(df)
df = df.groupby('col1')['col2'].apply(list)
print("\nGroup on the col1:")
print(df)

Пример вывода:

 Оригинальный фрейм данных
  col1 col2
0 С1 1
1 С1 2
2 С2 3
3 С2 3
4 С2 4
5 С3 6
6 С2 5
Группа по кол1:
col1
C1 [1, 2]
С2 [3, 3, 4, 5]
С3 [6]
Имя: col2, dtype: object       

Редактор кода Python:

Есть другой способ решить это решение? Внесите свой код (и комментарии) через Disqus.

Предыдущий: Напишите программу Pandas для переименования определенного имени столбца в заданном кадре данных.
Далее: Напишите программу Pandas для получения индекса столбца из имени столбца данного DataFrame.