Преобразование Dataframe в список List на основе имени столбца в Python - PullRequest
2 голосов
/ 16 мая 2019

Входной сигнал:
Dataframe

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

Выход:
Список списка

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

Ответы [ 2 ]

2 голосов
/ 16 мая 2019

Создать Series из lists с помощью groupby с GroupBy.apply и в последний раз преобразовать его в list:

L = df.groupby('col1')['col2'].apply(list).tolist()
print (L)
[[10, 20, 25], [56, 36], [1]]
1 голос
/ 16 мая 2019

Вы можете GroupBy col1 и извлекать списки из каждой группы по col2:

df.groupby('col1').col2.apply(list).values.tolist()
# [[10, 20, 25], [56, 36], [1]]
Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...